WIP

python/src/handlers/chat.py

@@ -23,19 +23,23 @@ def handler(event, context):
   }
   question = get_query(event)
   index_name = get_param(event, "index", "Work")
-  text_key = get_param(event, "text_key", "title")
-  attributes = get_param(event, "attributes", "identifier,title").split(",")
-  attributes.append("source")
+  text_key = get_param(event, "text_key", "title")  
+  attributes = get_param(event, 
+                         "attributes", 
+                         "identifier,title,source,alternate_title,contributor,create_date,creator,date_created,description,genre,keywords,language,location,physical_description_material,physical_description_size,scope_and_contents,style_period,subject,table_of_contents,technique").split(",")
 
   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())
+    retriever=weaviate.as_retriever(),
+    return_source_documents=True)
 
   response = chain({"question": question})
 
@@ -44,7 +48,7 @@ def handler(event, context):
     "headers": {
       "Content-Type": "application/json"
     },
-    "body": json.dumps(response)
+    "body": response
   }
 
 

python/src/setup.py

@@ -25,10 +25,7 @@ def weaviate_vector_store(index_name: str, text_key: str, attributes: List[str]
 
   client = weaviate.Client(
       url=weaviate_url,
-      auth_client_secret=auth_config,
-      additional_headers={
-          "X-OpenAI-Api-Key": openai_api_key
-      }
+      auth_client_secret=auth_config
   )
   return Weaviate(client=client, 
                   index_name=index_name,