Create keyword.py

keyword.py

@@ -0,0 +1,20 @@
+import openai
+
+def keyword_agent(query, key):
+
+    system_prompt = """
+    You are an AI assistant that looks at a given query about a school class, and comes up with a list of 3 keywords related to the query.
+        Two keywords will be used that are similar in semantics to the given query but not already used in query.
+        The third keyword should focus on an aspect related to the query that is less obvious, but might be helpful when finding similar data entries.
+    If the query does not seem to relate to a school class, please output nothing or just spaces " ".
+    """
+
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": query}
+        ]
+    )
+
+    return response["choices"][0]["message"]["content"]