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:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user