phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-05 12:21:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge remote-tracking branch 'origin/api_keys' into api_updates_2

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-17 14:31:22 -07:00
commit ed351b8276
5 changed files with 16 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

16
llama_stack/providers/impls/meta_reference/agents/agents.py
View file

@ -58,21 +58,9 @@ class MetaReferenceAgentsImpl(Agents):
builtin_tools = []
for tool_defn in agent_config.tools:
if isinstance(tool_defn, WolframAlphaToolDefinition):
key = self.config.wolfram_api_key
if not key:
raise ValueError("Wolfram API key not defined in config")
tool = WolframAlphaTool(key)
tool = WolframAlphaTool(tool_defn.api_key)
elif isinstance(tool_defn, SearchToolDefinition):
key = None
if tool_defn.engine == SearchEngineType.brave:
key = self.config.brave_search_api_key
elif tool_defn.engine == SearchEngineType.bing:
key = self.config.bing_search_api_key
if not key:
raise ValueError(
"Search (Brave or Bing) API key not defined in config"
)
tool = SearchTool(tool_defn.engine, key)
tool = SearchTool(tool_defn.engine, tool_defn.api_key)
elif isinstance(tool_defn, CodeInterpreterToolDefinition):
tool = CodeInterpreterTool()
elif isinstance(tool_defn, PhotogenToolDefinition):