Merge pull request #7 from bettersg/agent/dev/openai-assistants-api-1

added a few stuff
bettersg · Feb 25, 2024 · 115b62a · 115b62a
2 parents 85ff193 + 4010f9a
commit 115b62a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/implementation/agent.py b/implementation/agent.py
@@ -4,8 +4,12 @@
 from langchain.tools import Tool
 from langchain_community.utilities import GoogleSearchAPIWrapper
 import json
+from openai import OpenAI
+import os
 
-search = GoogleSearchAPIWrapper()
+search = GoogleSearchAPIWrapper(google_api_key=os.getenv("GOOGLE_SEARCH_API_KEY"))
+
+client = OpenAI()
 
 def top10(query):
     return search.results(query, 10)
@@ -27,3 +31,6 @@ def check_message(self, message):
 
     def search_google(self, q):
         return json.dumps(tool.run(q))
+
+    def agent_report(self, reasoning, category, truth_score = None, subjects = None) -> Vote:
+        return Vote(category=category, truthScore=truth_score)