Upload 7 files

generator.py

@@ -14,7 +14,7 @@ class RAGAgent:
             self,
             retriever=retriever,
             reranker=reranker,
-            anthropic_api_key: str = os.environ.getattribute("anthropic_api_key"),
+            anthropic_api_key: str = os.environ["anthropic_api_key"],
             model: str = "claude-3-5-sonnet-20241022",
             max_tokens: int = 1024,
             temperature: float = 0.0,

rag_agent.py

@@ -16,7 +16,7 @@ class RAGAgent:
             self,
             retriever=retriever,
             reranker=reranker,
-            anthropic_api_key: str = os.environ.getattribute("anthropic_api_key"),
+            anthropic_api_key: str = os.environ["anthropic_api_key"],
             model: str = "claude-3-5-sonnet-20241022",
             max_tokens: int = 1024,
             temperature: float = 0.0,

retriever.py

@@ -6,7 +6,7 @@ from text_embedder_encoder import TextEmbedder, encoder_model_name
 
 class Retriever:
     def __init__(self,
-                 pinecone_api_key=os.environ.getattribute("pinecone_api_key"),
+                 pinecone_api_key=os.environ["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)