Upload 7 files

hide credentials

generator.py

@@ -2,6 +2,7 @@ from retriever import Retriever
 from reranker import Reranker
 from anthropic import Anthropic
 from typing import List
+import os
 
 
 retriever = Retriever()
@@ -13,7 +14,7 @@ class RAGAgent:
             self,
             retriever=retriever,
             reranker=reranker,
-            anthropic_api_key: str = "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",
+            anthropic_api_key: str = os.environ.getattribute("anthropic_api_key"),
             model: str = "claude-3-5-sonnet-20241022",
             max_tokens: int = 1024,
             temperature: float = 0.0,

rag_agent.py

@@ -1,5 +1,6 @@
 from anthropic import Anthropic
 from typing import List
+import os
 
 
 from retriever import Retriever
@@ -15,7 +16,7 @@ class RAGAgent:
             self,
             retriever=retriever,
             reranker=reranker,
-            anthropic_api_key: str = "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",
+            anthropic_api_key: str = os.environ.getattribute("anthropic_api_key"),
             model: str = "claude-3-5-sonnet-20241022",
             max_tokens: int = 1024,
             temperature: float = 0.0,

retriever.py

@@ -1,12 +1,12 @@
 from pinecone import Pinecone
-
+import os
 
 from text_embedder_encoder import TextEmbedder, encoder_model_name
 
 
 class Retriever:
     def __init__(self,
-                 pinecone_api_key="pcsk_468XZz_QfKbP3dWCh6nLatJjd882DGF5HDh6TupzEAeRpFLAMtDfDiPDNRC537Q4jAtxhV",
+                 pinecone_api_key=os.environ.getattribute("pinecone_api_key"),
                  index_name=f"hebrew-dentist-qa-{encoder_model_name.replace('/', '-')}".lower()):
         # Initialize Pinecone connection
         self.pc = Pinecone(api_key=pinecone_api_key)