Commit 2

added SYSTEM_PROMPT and client.chat.completions.create(messages[])
testing prompt_templates

app.py

@@ -11,18 +11,18 @@ from Gradio_UI import GradioUI
 # Here we suppose that the textual description of the tools has already been appended
 SYSTEM_PROMPT = """Answer the following questions as best you can. You have access to the following tools:
 
-get_weather: Get the current weather in a given location
+get_current_time_in_timezone: Fetches the current local time in a specified timezone
 
 The way you use the tools is by specifying a json blob.
 Specifically, this json should have a `action` key (with the name of the tool to use) and a `action_input` key (with the input to the tool going here).
 
 The only values that should be in the "action" field are:
-get_weather: Get the current weather in a given location, args: {"location": {"type": "string"}}
+get_current_time_in_timezone: Get the current current local time in a specified timezone, args: {"timezone": {"type": "string"}}
 example use :
 ```
 {{
-  "action": "get_weather",
-  "action_input": {"location": "New York"}
+  "action": "get_current_time_in_timezone",
+  "action_input": {"timezone": 'America/New_York'}
 }}
 
 ALWAYS use the following format:
@@ -43,6 +43,17 @@ Final Answer: the final answer to the original input question
 
 Now begin! Reminder to ALWAYS use the exact characters `Final Answer:` when you provide a definitive answer. """
 
+output = client.chat.completions.create(
+    messages=[
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {"role": "agent", "content": "Hi, I am an agent that knows the current time..."},
+        {"role": "user", "content": "{input}"},
+    ],
+    stream=False,
+    max_tokens=2048,
+    stop=["Observation:"] # Let's stop before any actual function is called
+)
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -82,16 +93,19 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
+# new_prompt=prompt+output+get_current_time_in_timezone('{input}')
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
+
+print("yaml" + prompt_templates)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,