openai tools: TS signatures work well too at a fraction of the eval cost

Author: ochafik

examples/openai/prompting.py

@@ -608,7 +608,8 @@ def __init__(self, args: ChatHandlerArgs, parallel_calls: bool):
             content='\n'.join([
                 'You are a function calling AI model.',
                 'Here are the tools available:',
-                _tools_schema_signatures(self.args.tools, indent=2),
+                # _tools_schema_signatures(self.args.tools, indent=2),
+                _tools_typescript_signatures(self.args.tools),
                 _please_respond_with_schema(
                     _make_bespoke_schema(
                         response_schema,
@@ -713,9 +714,10 @@ def get_chat_handler(args: ChatHandlerArgs, parallel_calls: bool, tool_style: Op
 
 _ts_converter = SchemaToTypeScriptConverter()
 
+# os.environ.get('NO_TS')
 def _please_respond_with_schema(schema: dict) -> str:
-    sig = json.dumps(schema, indent=2)
-    # sig = _ts_converter.visit(schema)
+    # sig = json.dumps(schema, indent=2)
+    sig = _ts_converter.visit(schema)
     return f'Please respond in JSON format with the following schema: {sig}'
 
 def _tools_typescript_signatures(tools: list[Tool]) -> str: