Bugfix - date format

DEENUU1 · Jan 28, 2024 · 349b057 · 349b057
1 parent 0dba7e6
commit 349b057
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/app/ai/agent.py b/app/ai/agent.py
@@ -38,6 +38,7 @@ def setup_agent(session_id: str, model: str) -> AgentExecutor:
 
     tools = [
         google_calendar.GoogleCalendarCreateEventTool(),
+        google_calendar.GoogleCalendarListEventTool(),
         notion.NotionNoteCreateTool(),
         today.CurrentTimeTool(),
         news.NewsTool(),

diff --git a/app/ai/tools/google_calendar.py b/app/ai/tools/google_calendar.py
@@ -14,15 +14,15 @@ class GoogleCalendarCreateEventInput(BaseModel):
     )
     event_name: str = Field(description="Name of the event")
     all_day: bool = Field(description="If event is all day")
-    start_date: str = Field(description="Start date of the event in format 'YYYY-MM-DD HH:MM' or 'YYYY-MM-DD'")
-    end_date: Optional[str] = Field(description="End date of the event in format 'YYYY-MM-DD HH:MM' or 'YYYY-MM-DD'")
+    start_date: str = Field(description="Start date of the event in format 'YYYY/MM/DD HH:MM' or 'YYYY/MM/DD'")
+    end_date: Optional[str] = Field(description="End date of the event in format 'YYYY/MM/DD HH:MM' or 'YYYY/MM/DD'")
     duration: Optional[str] = Field(
         description="Duration of the event in format HH:mm use only when user gives use duration of an event"
     )
 
 
 class GoogleCalendarListEventInput(BaseModel):
-    start_date: str = Field(description="Start date of the event in format 'YYYY-MM-DD'")
+    start_date: str = Field(description="Start date of the event in format 'YYYY/MM/DD'")
 
 
 class GoogleCalendarCreateEventTool(BaseTool):
@@ -87,7 +87,7 @@ def get_events_from_all_calendars(start_date: str) -> List:
                     "operation": "list"  # Hard coded value based on routing in make.com,
                 }
             )
-            result.append(response.json())
+            result.append(response.text)
 
         return result