phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

conditional import async_generator

Browse source
This commit is contained in:
ishaan-jaff 2023-09-21 11:01:51 -07:00
parent b88145e0b1
commit 6bfde2496c
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/ollama.py
View file

@ -1,6 +1,10 @@
import requests
import json
from async_generator import async_generator, yield_
try:
from async_generator import async_generator, yield_ # optional dependancy if you want to use acompletion + streaming
except:
pass # this should not throw an error, it will impact the 'import litellm' statement
# ollama implementation
def get_ollama_response_stream(