npc0 commited on
Commit
6989a20
1 Parent(s): ace2a93

Add today date to query

Browse files
Files changed (1) hide show
  1. app.py +18 -6
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import tempfile
 
2
  import gradio as gr
3
  import janus_swi as janus
4
  from crewai import Agent, Task, Crew
@@ -6,11 +7,21 @@ from crewai_tools import tool
6
  from crewai_tools import MDXSearchTool
7
  from crewai_tools import WebsiteSearchTool
8
  from langchain_anthropic import ChatAnthropic
9
- # import asyncio
10
- # loop = asyncio.new_event_loop()
11
- # asyncio.set_event_loop(loop)
12
- # import nest_asyncio
13
- # nest_asyncio.apply()
 
 
 
 
 
 
 
 
 
 
14
 
15
  DOC_URL = 'https://secure.ssa.gov/apps10/poms.nsf/lnx/0500502100'
16
  MODEL_NAME = "claude-3-5-sonnet-20240620"
@@ -142,7 +153,8 @@ crew = Crew(
142
 
143
 
144
  def yes_man(user_query, history):
145
- return crew.kickoff(inputs={"query": user_query}).raw
 
146
 
147
  gr.ChatInterface(
148
  yes_man,
 
1
  import tempfile
2
+ import datetime
3
  import gradio as gr
4
  import janus_swi as janus
5
  from crewai import Agent, Task, Crew
 
7
  from crewai_tools import MDXSearchTool
8
  from crewai_tools import WebsiteSearchTool
9
  from langchain_anthropic import ChatAnthropic
10
+
11
+
12
+ def add_today_date_to_string(input_string: str) -> str:
13
+ """Adds today's date to the given string in a natural language format.
14
+
15
+ Args:
16
+ input_string (str): The original string to which the date will be added.
17
+
18
+ Returns:
19
+ str: The modified string with today's date added in a natural language format.
20
+ """
21
+ today = datetime.date.today()
22
+ formatted_date = today.strftime("%B %d, %Y")
23
+ return f"# Date\nToday is {formatted_date}.\n# Query\n{input_string}"
24
+
25
 
26
  DOC_URL = 'https://secure.ssa.gov/apps10/poms.nsf/lnx/0500502100'
27
  MODEL_NAME = "claude-3-5-sonnet-20240620"
 
153
 
154
 
155
  def yes_man(user_query, history):
156
+ query = add_today_date_to_string(user_query)
157
+ return crew.kickoff(inputs={"query": query}).raw
158
 
159
  gr.ChatInterface(
160
  yes_man,