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

Several smaller fixes to make adapters work

Browse source 
Also, reorganized the pattern of __init__ inside providers so
configuration can stay lightweight
This commit is contained in:
Ashwin Bharambe 2024-08-28 09:42:08 -07:00
parent 2a1552a5eb
commit 45987996c4
23 changed files with 164 additions and 160 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_toolchain/agentic_system/client.py
View file

@ -194,11 +194,6 @@ async def run_rag(host: str, port: int):
MemoryToolDefinition(
max_tokens_in_context=2048,
memory_bank_configs=[],
# memory_bank_configs=[
# AgenticSystemVectorMemoryBankConfig(
# bank_id="970b8790-268e-4fd3-a9b1-d0e597e975ed",
# )
# ],
),
]
@ -210,8 +205,9 @@ async def run_rag(host: str, port: int):
await _run_agent(api, tool_definitions, user_prompts, attachments)
def main(host: str, port: int):
asyncio.run(run_rag(host, port))
def main(host: str, port: int, rag: bool = False):
fn = run_rag if rag else run_main
asyncio.run(fn(host, port))
if __name__ == "__main__":