kk

ask.py

@@ -58,16 +58,16 @@ def getBrains(name):
         return False
 
 
-def runAgent(memory, question, temperature, maxTokens):
+def runAgent(brainName,memory, question, temperature, maxTokens):
     if (memory == False):
         return "Please Initiate the Chat first.."
     temperature = float(temperature)
     pinecone.init(api_key=pinecone_key,
                   environment="us-west4-gcp")
-    pineconeindex = pinecone.Index("dimetyd-test")
+    pineconeindex = pinecone.Index(brainName)
     index = GPTPineconeIndex([], pinecone_index=pineconeindex)
     # memory = ConversationBufferMemory(memory_key="chat_history")
-    # print(memory.chat_memory)
+    print(memory.chat_memory)
     llm = OpenAI(
         temperature=temperature, model_name="text-davinci-003", max_tokens=maxTokens)
     tool_config = IndexToolConfig(
@@ -83,5 +83,5 @@ def runAgent(memory, question, temperature, maxTokens):
     agent_chain = create_llama_chat_agent(
         toolkit, llm, memory=memory, verbose=True)
     response = agent_chain.run(question)
-    # print(memory.chat_memory)
+    print(memory.chat_memory)
     return response, memory