Merge pull request #187 from RussellLuo/improve-runtime-config

feat(skills): add support for setting `api_key` and `api_base` in RuntimeConfig
This commit is contained in:
RichardTang-Aden
2026-01-25 16:45:21 -08:00
committed by GitHub
3 changed files with 14 additions and 2 deletions
@@ -520,6 +520,8 @@ class RuntimeConfig:
model: str = "cerebras/zai-glm-4.7"
temperature: float = 0.7
max_tokens: int = 4096
api_key: str | None = None
api_base: str | None = None
default_config = RuntimeConfig()
@@ -972,7 +974,11 @@ class {agent_class_name}:
llm = None
if not mock_mode:
# LiteLLMProvider uses environment variables for API keys
llm = LiteLLMProvider(model=self.config.model)
llm = LiteLLMProvider(
model=self.config.model,
api_key=self.config.api_key,
api_base=self.config.api_base,
)
self._graph = GraphSpec(
id="{agent_name}-graph",
@@ -233,7 +233,11 @@ class OnlineResearchAgent:
llm = None
if not mock_mode:
# LiteLLMProvider uses environment variables for API keys
llm = LiteLLMProvider(model=self.config.model)
llm = LiteLLMProvider(
model=self.config.model,
api_key=self.config.api_key,
api_base=self.config.api_base,
)
self._graph = GraphSpec(
id="online-research-agent-graph",
@@ -7,6 +7,8 @@ class RuntimeConfig:
model: str = "groq/moonshotai/kimi-k2-instruct-0905"
temperature: float = 0.7
max_tokens: int = 16384
api_key: str | None = None
api_base: str | None = None
default_config = RuntimeConfig()