From c443751c7d1185168e97eccf5b518a8e1210dc3e Mon Sep 17 00:00:00 2001
From: Aaron White <awhite@tarvold.com>
Date: Wed, 28 Aug 2024 15:55:14 -0400
Subject: [PATCH] API update

Changed to invoke and include open.api_key
---
 query_data.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/query_data.py b/query_data.py
index 43ed9a5e6..b80cf9bab 100644
--- a/query_data.py
+++ b/query_data.py
@@ -5,7 +5,13 @@
 from langchain_openai import ChatOpenAI
 from langchain.prompts import ChatPromptTemplate
 
+import openai 
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
 CHROMA_PATH = "chroma"
+openai.api_key = os.environ["OPENAI_API_KEY"]
 
 PROMPT_TEMPLATE = """
 Answer the question based only on the following context:
@@ -41,7 +47,7 @@ def main():
     print(prompt)
 
     model = ChatOpenAI()
-    response_text = model.predict(prompt)
+    response_text = model.invoke(prompt)
 
     sources = [doc.metadata.get("source", None) for doc, _score in results]
     formatted_response = f"Response: {response_text}\nSources: {sources}"