From 349b057a4e621650c9109757fa04ce5d5c282b85 Mon Sep 17 00:00:00 2001 From: DEENUU1 Date: Sun, 28 Jan 2024 21:18:59 +0100 Subject: [PATCH] Bugfix - date format --- app/ai/agent.py | 1 + app/ai/tools/google_calendar.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/app/ai/agent.py b/app/ai/agent.py index 36b2490..253c1c5 100644 --- 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 index 13362ba..9b3c50b 100644 --- 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