PM-pinou
24 lines
994 B
Python
24 lines
994 B
Python
"""Test 32B-optimized LLM orchestrator"""
|
|
import sys, os, json
|
|
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 tianxuan.llm_orchestrator import run_llm_pipeline, LLMConfig
|
|
from analysis.session_store import SessionStore
|
|
from analysis.data_loader import load_csv_directory
|
|
|
|
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})
|
|
|
|
cfg = LLMConfig(base_url='https://api.deepseek.com',
|
|
api_key='sk-360ef76d59674d6b8bc2eb160327dd39',
|
|
model='deepseek-v4-flash')
|
|
|
|
result = run_llm_pipeline('ds', config=cfg)
|
|
print(f'Steps: {result.get("steps")}')
|
|
out = result.get("result", result.get("error", ""))
|
|
print(f'Result: {str(out)[:200]}')
|