From 4010f9a900ca11282c6f71597d6100bd61c3768a Mon Sep 17 00:00:00 2001
From: Bing Wen Tan <bingwen.cpf@gmail.com>
Date: Sun, 25 Feb 2024 18:28:40 +0800
Subject: [PATCH] added a few stuff

---
 implementation/agent.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/implementation/agent.py b/implementation/agent.py
index 2e50dc3..1bb9861 100644
--- 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)
\ No newline at end of file