modifying environment handling

agent.py

@@ -14,10 +14,10 @@ from langchain.agents.react.base import DocstoreExplorer
 from langchain import LLMMathChain
 from typing import Union
 from langchain.memory import ConversationBufferWindowMemory
-
+import os
 
 load_dotenv()
-config = dotenv_values(".env")
+#os.environ = dotenv_values(".env")
 
 
 
@@ -42,23 +42,23 @@ class QMLAgent():
 
     def __init__(self):
 
-        pinecone.init(api_key=config["PINECONE_API_KEY"], environment=config["PINECONE_REGION"])
+        pinecone.init(api_key=os.environ["PINECONE_API_KEY"], environment=os.environ["PINECONE_REGION"])
 
-        index = pinecone.Index(config["INDEX_NAME"])
+        index = pinecone.Index(os.environ["INDEX_NAME"])
 
         embeddings = OpenAIEmbeddings()
 
 
         bm25_encoder = BM25Encoder()
-        bm25_encoder.load(config["BM25_FILENAME"])
+        bm25_encoder.load(os.environ["BM25_FILENAME"])
 
         retriever = PineconeHybridSearchRetriever(
             embeddings=embeddings, 
             sparse_encoder=bm25_encoder, 
             index=index, 
-            top_k=config["TOP_K"])
+            top_k=os.environ["TOP_K"])
 
-        llm = ChatOpenAI(model_name=config["CHAT_MODEL"])
+        llm = ChatOpenAI(model_name=os.environ["CHAT_MODEL"])
 
         math_tool = CalculatorTool()
 
@@ -81,7 +81,7 @@ class QMLAgent():
             ),
         ]
 
-        memory = ConversationBufferWindowMemory(k=config["MEMORY_LENGTH"], memory_key="chat_history", return_messages=True)
+        memory = ConversationBufferWindowMemory(k=os.environ["MEMORY_LENGTH"], memory_key="chat_history", return_messages=True)
 
 
         self.agent_chain = initialize_agent(