make input similar to example

aurelio-labs · Jan 11, 2024 · 19984ff · 19984ff
1 parent 0a0a205
commit 19984ff
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/semantic_router/utils/function_call.py b/semantic_router/utils/function_call.py
@@ -64,15 +64,17 @@ def extract_function_inputs(
     "output": "<class 'float'>",
 }}
 
-
-Result: {{
+Result:
+{{
     "location": "Hawaii",
     "degree": "Kelvin",
 }}
 
 Input:
-query: {query}
-schema: {function_schema}
+query: \"{query}\"
+schema:
+{json.dumps(function_schema, indent=4)}
+
 Result:
 """
     llm_input = [Message(role="user", content=prompt)]