import os
import re
import requests
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI

load_dotenv()

llm = ChatOpenAI(
    model="qwen3.5-35b-a3b",
    openai_api_key=os.getenv("AGICTO_API_KEY"),
    openai_api_base=os.getenv("AGICTO_BASE_URL", "https://api.agicto.cn/v1"),
    temperature=0.0,
    max_tokens=256,
)

SYSTEM_PROMPT = (
    "Answer with one word or number.\n"
    "No explanation.\n"
    "No punctuation.\n"
    "No units.\n"
    "No tags."
)

def clean(text: str) -> str:
    text = re.sub(r".*?", "", text, flags=re.S)
    return text.strip().splitlines()[0]

def agent(question: str, files=None) -> str:
    try:
        context = ""
        if files:
            for f in files:
                r = requests.get(f, timeout=10)
                if r.ok:
                    context += r.text[:6000]

        res = llm.invoke([
            {"role": "system", "content": SYSTEM_PROMPT},
            {"role": "user", "content": f"{context}{question}"}
        ])

        return clean(res.content)

    except Exception:
        return "0"