"""
python -m src.proof_of_concepts.run_extract_one_user
"""
import json
from typing import List, Optional
from pydantic import BaseModel
from src.llm_factory import get_llm

class User(BaseModel):
    id: int
    name: str = "Jane Doe"

llm = get_llm("ollama-llama3.1")
# llm = get_llm("openrouter-paid-gemini-2.0-flash-001")
sllm = llm.as_structured_llm(User)

text = "location=unspecified, user id=42, role=agent, name=Simon, age=30"

response = sllm.complete(text)

json_response = json.loads(response.text)
print(json.dumps(json_response, indent=2))