Update app.py

app.py

@@ -452,21 +452,25 @@ weather_api_key = os.environ.get("WEATHER_API_KEY")
 
 def get_time_tool2(query: str) -> datetime:
     try:
-        # Step 1: Get location from the query
+        # Step 1: 獲取地點
         location_prompt = f"""
         You are a location extractor. Given a user's query about time or date, return the location mentioned in it.
         If not found, return "London".
         
         Query: "{query}"
         """
-        location_response = llm_gpt4.invoke(location_prompt)
-        location = location_response.content.strip() if isinstance(location_response, AIMessage) else str(location_response).strip()
+        location_response = openai.Completion.create(
+            model="text-davinci-003",
+            prompt=location_prompt,
+            max_tokens=50
+        )
+        location = location_response.choices[0].text.strip()
 
-        # Step 2: Get the timezone of the location
-        tz_str = location_to_timezone(location)
-        now = datetime.now(ZoneInfo(tz_str))
+        # Step 2: 獲取當前時間（當地時間）
+        tz_str = location_to_timezone(location)  # 從地點獲取時區
+        now = datetime.now(ZoneInfo(tz_str))  # 當地時間
 
-        # Step 3: Use GPT to determine the specific time
+        # Step 3: 用 GPT 推算具體時間
         time_query_prompt = f"""
         Given the current time in {location}, answer the following:
 
@@ -474,32 +478,28 @@ def get_time_tool2(query: str) -> datetime:
 
         Provide the exact time the user is referring to. If the query is asking for a time in the future or past, calculate that time relative to now.
         """
-        time_response = llm_gpt4.invoke(time_query_prompt)
+        time_response = openai.Completion.create(
+            model="text-davinci-003",
+            prompt=time_query_prompt,
+            max_tokens=50
+        )
+        time_str = time_response.choices[0].text.strip()
 
-        if isinstance(time_response, AIMessage):
-            time_str = time_response.content.strip()
+        # Step 4: 解析並格式化時間
+        # 嘗試解析時間
+        if "AM" in time_str or "PM" in time_str:
+            target_time = datetime.strptime(time_str, "%I:%M %p")
         else:
-            time_str = str(time_response).strip()
-
-        # Step 4: If the time string contains specific time formats, adjust accordingly
-        try:
-            # Check if time string contains AM/PM
-            if "AM" in time_str or "PM" in time_str:
-                target_time = datetime.strptime(time_str, "%I:%M %p")
-            else:
-                # Otherwise, parse the date and time
-                target_time = datetime.strptime(time_str, "%Y-%m-%d %H:%M:%S")
+            target_time = datetime.strptime(time_str, "%Y-%m-%d %H:%M:%S")
 
-            # Ensure it's in the correct timezone
-            target_time = target_time.replace(tzinfo=ZoneInfo(tz_str))
+        # 添加時區
+        target_time = target_time.replace(tzinfo=ZoneInfo(tz_str))
 
-            return target_time.strftime("%Y-%m-%d %H:%M:%S")
-        except Exception as e:
-            return f"Error in processing time: {e}"
+        return target_time.strftime("%Y-%m-%d %H:%M:%S")
 
     except Exception as e:
-        return f"Error in retrieving location or time information: {e}"
-
+        return f"Error in processing time: {e}"
+        
 def weather_agent_tool(query: str) -> str:
     """Weather Agent: Return current, hourly, or historical weather info using WeatherAPI."""
     try: