Spaces:
Running on Zero
Running on Zero
| import sys | |
| import asyncio | |
| from pathlib import Path | |
| # Add src to path | |
| sys.path.insert(0, str(Path("src").absolute())) | |
| from sunnyware.tools import _TOOLS | |
| from sunnyware.needle_tool_loader import build_schema_from_tool | |
| from sunnyware import local_llm | |
| async def main(): | |
| print(f"[1] Building schemas for {len(_TOOLS)} tools...") | |
| all_schemas = [build_schema_from_tool(tool) for tool in _TOOLS.values()] | |
| print(f"[2] Built {len(all_schemas)} schemas.") | |
| prompt = "Use Python to calculate the first 10 fibonacci numbers." | |
| system = "You are a helpful assistant." | |
| print("[3] Calling local LLM (this may take a few minutes)...") | |
| res = await local_llm.chat( | |
| prompt=prompt, | |
| system=system, | |
| history=[], | |
| tools=all_schemas, | |
| timeout=300.0 | |
| ) | |
| print("\n=== FINAL RESULT ===") | |
| print(f"ok: {res.get('ok')}") | |
| print(f"error: {res.get('error')}") | |
| content = res.get('content') | |
| if content: | |
| print(f"content (first 500 chars): {content[:500]}") | |
| else: | |
| print("content: None") | |
| tool_calls = res.get('tool_calls') | |
| print(f"tool_calls: {tool_calls}") | |
| if __name__ == "__main__": | |
| asyncio.run(main()) | |