test2

app.py

@@ -1,29 +1,62 @@
 from openai import OpenAI
 import os
+import math
+import datetime
+import requests
 
-class BasicAgent:
+class ToolEnhancedAgent:
     def __init__(self):
         api_key = os.getenv("OPENAI_API_KEY")
         if not api_key:
-            raise ValueError("OPENAI_API_KEY environment variable not set.")
+            raise ValueError("OPENAI_API_KEY is not set.")
         self.client = OpenAI(api_key=api_key)
-        print("BasicAgent (GPT-based) initialized.")
+        print("ToolEnhancedAgent initialized with GPT + CoT + Tools.")
+
+    def use_tool(self, tool_name: str, input_text: str) -> str:
+        try:
+            if tool_name == "calculator":
+                return str(eval(input_text))
+            elif tool_name == "date":
+                return str(datetime.datetime.now().date())
+            elif tool_name == "wikipedia":
+                summary = self.search_wikipedia(input_text)
+                return summary
+            else:
+                return "[Tool Error: Unknown Tool]"
+        except Exception as e:
+            return f"[Tool Error: {e}]"
+
+    def search_wikipedia(self, query):
+        try:
+            res = requests.get(f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}")
+            if res.status_code == 200:
+                return res.json().get("extract", "No summary found.")
+            else:
+                return f"No Wikipedia summary for {query}."
+        except Exception as e:
+            return f"Wiki error: {e}"
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        prompt = (
+            "You are a helpful AI assistant. Use tools when necessary. "
+            "Think step-by-step before answering. Respond clearly.\n\n"
+            f"Question: {question}\n"
+            "Answer (show thinking steps):"
+        )
+
         try:
             response = self.client.chat.completions.create(
-                model="gpt-4",  # Bisa diganti dengan "gpt-3.5-turbo" jika quota terbatas
+                model="gpt-4",
                 messages=[
-                    {"role": "system", "content": "You are a helpful assistant."},
-                    {"role": "user", "content": question}
+                    {"role": "system", "content": "You are a smart assistant that can use tools and think step-by-step."},
+                    {"role": "user", "content": prompt}
                 ],
-                temperature=0.7,
-                max_tokens=500,
+                temperature=0.3,
+                max_tokens=700,
             )
             answer = response.choices[0].message.content.strip()
-            print(f"Agent returning answer: {answer[:100]}...")
+            print(f"Answer generated: {answer[:100]}...")
             return answer
         except Exception as e:
-            print(f"Error during OpenAI completion: {e}")
-            return f"[ERROR from agent: {str(e)}]"
+            print(f"[Agent Error]: {e}")
+            return f"[Agent Error: {e}]"