69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
"""Cursor SDK 调用（daily 包专用）。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
|
||
from cursor_sdk import Agent, AgentOptions, Client, CursorAgentError, LocalAgentOptions
|
||
|
||
from daily.config import env, env_int
|
||
from daily.cursor_bridge import cursor_cwd, warm_cursor_bridge
|
||
|
||
_sdk_client: Client | None = None
|
||
_sdk_client_key: tuple[str, str, float] | None = None
|
||
|
||
|
||
def cursor_timeout_seconds() -> float:
|
||
"""Cursor bridge unary/stream 超时（秒）；与 bot 共用 CURSOR_TIMEOUT。"""
|
||
return float(env_int("CURSOR_TIMEOUT", 600))
|
||
|
||
|
||
def _cursor_sdk_client() -> Client:
|
||
"""带自定义超时的 bridge Client（SDK 默认 unary 仅 60s，复杂 Agent 任务易超时）。"""
|
||
global _sdk_client, _sdk_client_key
|
||
|
||
warm_cursor_bridge()
|
||
url = os.environ.get("CURSOR_SDK_BRIDGE_URL", "")
|
||
token = os.environ.get("CURSOR_SDK_BRIDGE_TOKEN", "")
|
||
timeout = cursor_timeout_seconds()
|
||
key = (url, token, timeout)
|
||
if _sdk_client is not None and _sdk_client_key == key:
|
||
return _sdk_client
|
||
if _sdk_client is not None:
|
||
_sdk_client.close()
|
||
_sdk_client = Client(
|
||
base_url=url,
|
||
auth_token=token,
|
||
unary_timeout=timeout,
|
||
stream_timeout=timeout,
|
||
allow_api_key_env_fallback=False,
|
||
)
|
||
_sdk_client_key = key
|
||
return _sdk_client
|
||
|
||
|
||
def cursor_chat(system: str, user: str) -> str:
|
||
api_key = (env("CURSOR_API_KEY") or "").strip()
|
||
if not api_key:
|
||
return ""
|
||
|
||
client = _cursor_sdk_client()
|
||
cwd = cursor_cwd()
|
||
model = env("CURSOR_MODEL") or "composer-2.5"
|
||
prompt = f"{system}\n\n{user}"
|
||
try:
|
||
result = Agent.prompt(
|
||
prompt,
|
||
AgentOptions(
|
||
api_key=api_key,
|
||
model=model,
|
||
local=LocalAgentOptions(cwd=cwd),
|
||
),
|
||
client=client,
|
||
)
|
||
except CursorAgentError as exc:
|
||
raise RuntimeError(f"LLM 调用失败：{exc.message}") from exc
|
||
if result.status == "error":
|
||
raise RuntimeError(f"LLM 调用失败：{result.result or '未知错误'}")
|
||
return (result.result or "").strip()
|