dekun
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
"""中控 AI 模型调用（共用 ai_client 配置，逻辑独立）。"""
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Optional, Sequence
|
|
|
|
_REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
if str(_REPO_ROOT) not in sys.path:
|
|
sys.path.insert(0, str(_REPO_ROOT))
|
|
|
|
from ai_client import ai_generate, ai_generate_chat, ai_provider_label # noqa: E402
|
|
|
|
|
|
def model_label() -> str:
|
|
return ai_provider_label()
|
|
|
|
|
|
def generate_text(
|
|
*,
|
|
system: str,
|
|
user: str,
|
|
temperature: float,
|
|
images_b64: Optional[Sequence[str]] = None,
|
|
max_tokens: int | None = None,
|
|
max_continuations: int = 3,
|
|
) -> str:
|
|
if max_tokens is not None and max_tokens > 0:
|
|
return ai_generate_chat(
|
|
system=system,
|
|
user=user,
|
|
temperature=temperature,
|
|
images_b64=images_b64,
|
|
max_tokens=int(max_tokens),
|
|
max_continuations=max_continuations,
|
|
)
|
|
prompt = f"{system.strip()}\n\n---\n\n{user.strip()}"
|
|
return ai_generate(
|
|
prompt,
|
|
temperature=temperature,
|
|
images_b64=images_b64,
|
|
)
|