WIP

python/src/handlers/chat.py

@@ -2,6 +2,8 @@
 import json
 import os
 import setup
+from langchain.chains import RetrievalQAWithSourcesChain
+
 
 from langchain.agents.agent_toolkits import (
     create_vectorstore_agent,
@@ -28,17 +30,14 @@ def handler(event, context):
   weaviate = setup.weaviate_vector_store(index_name=index_name, 
                                          text_key=text_key, 
                                          attributes=attributes)
+
+  client = setup.openai_chat_client()
+  chain = RetrievalQAWithSourcesChain.from_chain_type(
+    client, 
+    chain_type="stuff", 
+    retriever=weaviate.as_retriever())
 
-  vectorstore_info = VectorStoreInfo(
-      name=index_name,
-      description="NULDC Works",
-      vectorstore=weaviate,
-  )
-  llm = setup.openai_chat_client()
-  toolkit = VectorStoreToolkit(vectorstore_info=vectorstore_info, llm=llm)
-  agent_executor = create_vectorstore_agent(llm=llm, toolkit=toolkit, verbose=True)
-  print(f"QUESTION: {question}")
-  response = agent_executor.run(question)
+  response = chain({"question": question})
 
   return {
     "statusCode": 200,