Upload get_keywords.py

get_keywords.py

@@ -1,13 +1,12 @@
-from langchain_openai import ChatOpenAI
-from langchain_core.messages import (
-    HumanMessage,
-    SystemMessage
-)
-
+from openai import OpenAI
 from rake_nltk import Rake
 import nltk
 nltk.download('stopwords')
 nltk.download('punkt')
+
+# Initialize OpenAI client
+client = OpenAI()
+
 """
 This function takes in user query and returns keywords
 Input:
@@ -52,12 +51,16 @@ Input:
 Output: keywords: str
 """
 def get_keywords_openai(user_query: str) -> str:
-    llm = ChatOpenAI(temperature=0.0)
-    command = "return the keywords of the following query. response should be words separated by commas. "
-    message = [
-        SystemMessage(content=command),
-        HumanMessage(content=user_query)
+    messages = [
+        {"role": "system", "content": "return the keywords of the following query. response should be words separated by commas."},
+        {"role": "user", "content": user_query}
     ]
-    response = llm(message)
-    res = response.content.replace(",", "")
+    
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=messages,
+        temperature=0
+    )
+    response = completion.choices[0].message.content
+    res = response.replace(",", "")
     return res