liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
"""Configuration for Active Tool Selection Agent."""
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# LLM Configuration
|
|
LLM_PROVIDER = os.getenv("LLM_PROVIDER", "openai").lower()
|
|
LLM_PROVIDER = {"qwen": "dashscope", "bailian": "dashscope"}.get(LLM_PROVIDER, LLM_PROVIDER)
|
|
if LLM_PROVIDER == "dashscope":
|
|
OPENAI_API_KEY = os.getenv("DASHSCOPE_API_KEY")
|
|
OPENAI_BASE_URL = os.getenv(
|
|
"DASHSCOPE_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"
|
|
)
|
|
OPENAI_MODEL = os.getenv("DASHSCOPE_MODEL", "qwen3.7-plus")
|
|
else:
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
|
|
OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-5.6-luna")
|
|
|
|
|
|
def _map_model_for_openrouter(model: str) -> str:
|
|
"""Map a plain model id onto OpenRouter's `provider/model` form.
|
|
|
|
Ids that already contain "/" pass through unchanged; gpt-*/o1-*/o3-*/o4-*
|
|
become openai/…; claude-* becomes anthropic/claude-opus-4.8.
|
|
"""
|
|
if "/" in model:
|
|
return model
|
|
m = model.lower()
|
|
if m.startswith(("gpt-", "o1-", "o3-", "o4-")):
|
|
return f"openai/{model}"
|
|
if m.startswith("claude-"):
|
|
return "anthropic/claude-opus-4.8"
|
|
return model
|
|
|
|
|
|
# Universal fallback + gpt-5.x preference: route through OpenRouter when no direct
|
|
# OPENAI_API_KEY is configured, OR when the model is a gpt-5.x id (incl. gpt-5.6*)
|
|
# which needs OpenAI org-verification on the direct API. Explicit OPENAI_BASE_URL /
|
|
# OPENAI_MODEL overrides are kept.
|
|
_OR_KEY = os.getenv("OPENROUTER_API_KEY")
|
|
if _OR_KEY and (not OPENAI_API_KEY or OPENAI_MODEL.lower().startswith("gpt-5")):
|
|
OPENAI_API_KEY = _OR_KEY
|
|
if not os.getenv("OPENAI_BASE_URL"):
|
|
OPENAI_BASE_URL = "https://openrouter.ai/api/v1"
|
|
OPENAI_MODEL = _map_model_for_openrouter(OPENAI_MODEL)
|
|
|
|
# Agent Configuration
|
|
AGENT_TEMPERATURE = 0.7
|
|
MAX_TOOL_REQUESTS = 5 # Maximum number of tool discovery iterations
|
|
|
|
# Semantic Routing Configuration
|
|
SIMILARITY_THRESHOLD = 0.15 # Minimum similarity score for tool matching
|
|
TOP_K_SERVERS = 3 # Number of top servers to search
|
|
TOP_K_TOOLS = 5 # Number of top tools to return per server
|