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

simplify search tool and enable configuration for search engine

Browse source
This commit is contained in:
Hardik Shah 2024-09-09 18:41:11 -07:00
parent 640c5f8ab9
commit bdede6d14e
6 changed files with 56 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

18
llama_toolchain/agentic_system/client.py
View file

@ -16,7 +16,6 @@ from pydantic import BaseModel
from termcolor import cprint
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_toolchain.common.deployment_types import RestAPIExecutionConfig
from llama_toolchain.core.datatypes import RemoteProviderConfig
from .api import * # noqa: F403
@ -135,21 +134,8 @@ async def run_main(host: str, port: int):
api = AgenticSystemClient(f"http://{host}:{port}")
tool_definitions = [
BraveSearchToolDefinition(
remote_execution=RestAPIExecutionConfig(
url=URL(uri="https://api.bing.microsoft.com/v7.0/search"),
method=RestAPIMethod.GET,
headers={
"Ocp-Apim-Subscription-Key": "2259b3f9e0cb4fc9b968bb3b02ab13e7"
},
params={
"count": 3,
"textDecorations": True,
"textFormat": "HTML",
},
)
),
# WolframAlphaToolDefinition(),
SearchToolDefinition(engine=SearchEngineType.bing),
WolframAlphaToolDefinition(),
CodeInterpreterToolDefinition(),
]
tool_definitions += [