phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Support data: in URL for memory. Add ootb support for pdfs (#67)

Browse source 
* support data: in URL for memory. Add ootb support for pdfs

* moved utility to common and updated data_url parsing logic

---------

Co-authored-by: Hardik Shah <hjshah@fb.com>
This commit is contained in:
Hardik Shah 2024-09-12 13:00:21 -07:00 • committed by GitHub
parent 736092f6bc
commit 1d0e91d802
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 112 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_toolchain/agentic_system/client.py
View file

@ -83,13 +83,12 @@ class AgenticSystemClient(AgenticSystem):
if line.startswith("data:"):
data = line[len("data: ") :]
try:
if "error" in data:
jdata = json.loads(data)
if "error" in jdata:
cprint(data, "red")
continue
yield AgenticSystemTurnResponseStreamChunk(
**json.loads(data)
)
yield AgenticSystemTurnResponseStreamChunk(**jdata)
except Exception as e:
print(data)
print(f"Error with parsing or validation: {e}")