PM-pinou
50 lines
2.5 KiB
Python
50 lines
2.5 KiB
Python
"""Debug LLM orchestrator 400 on second call"""
|
|
import sys, os, json, asyncio
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'tianxuan.settings'
|
|
import django; django.setup()
|
|
from analysis.session_store import SessionStore
|
|
from analysis.data_loader import load_csv_directory
|
|
from analysis.tool_registry import handle_call
|
|
import urllib.request
|
|
|
|
store = SessionStore(); store.drop_all()
|
|
csv = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'data', 'complex_test.csv')
|
|
lf, schema, rc, fc, mem = load_csv_directory(csv)
|
|
store.store_dataset('ds', lf, schema=schema, metadata={'row_count': rc, 'csv_glob': csv})
|
|
|
|
TOOLS = [{"type": "function", "function": {"name": "profile_data",
|
|
"description": "Get column statistics",
|
|
"parameters": {"type": "object", "properties": {"dataset_id": {"type": "string"}}, "required": ["dataset_id"]}}}]
|
|
|
|
# Step 1: Get tool call from LLM
|
|
msgs = [{"role": "user", "content": "Call profile_data with ds"}]
|
|
payload = json.dumps({"model": "deepseek-v4-flash", "messages": msgs, "tools": TOOLS,
|
|
"tool_choice": "auto", "max_tokens": 200}).encode()
|
|
req = urllib.request.Request('https://api.deepseek.com/chat/completions', data=payload,
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer sk-360ef76d59674d6b8bc2eb160327dd39'},
|
|
method='POST')
|
|
resp = json.loads(urllib.request.urlopen(req, timeout=10).read())
|
|
msg = resp["choices"][0]["message"]
|
|
tc = msg["tool_calls"][0]
|
|
print(f'Got tool call: {tc["function"]["name"]}')
|
|
|
|
# Execute tool
|
|
result = asyncio.run(handle_call(tc["function"]["name"], json.loads(tc["function"]["arguments"])))
|
|
content = json.dumps(result, default=str, ensure_ascii=False)
|
|
print(f'Result length: {len(content)}')
|
|
|
|
# Step 2: Send back - with correct message format
|
|
msgs2 = msgs + [msg, {"role": "tool", "tool_call_id": tc["id"], "content": content[:2000]}]
|
|
payload2 = json.dumps({"model": "deepseek-v4-flash", "messages": msgs2, "tools": TOOLS,
|
|
"tool_choice": "auto", "max_tokens": 200}).encode()
|
|
req2 = urllib.request.Request('https://api.deepseek.com/chat/completions', data=payload2,
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer sk-360ef76d59674d6b8bc2eb160327dd39'},
|
|
method='POST')
|
|
try:
|
|
resp2 = json.loads(urllib.request.urlopen(req2, timeout=10).read())
|
|
print(f'Step 2 OK: {resp2["choices"][0]["message"]["content"][:100]}')
|
|
except urllib.error.HTTPError as e:
|
|
body = e.read().decode()[:500]
|
|
print(f'Step 2 400: {body}')
|