add more tests

This commit is contained in:
Adil Hafeez 2024-10-27 15:02:43 -07:00
parent f5ecf733ff
commit e35560623c
7 changed files with 171 additions and 49 deletions

View file

@ -12,6 +12,7 @@ LLM_GATEWAY_ENDPOINT = os.getenv(
def get_data_chunks(stream, n=1):
chunks = []
for chunk in stream.iter_lines():
print(chunk)
if chunk:
chunk = chunk.decode("utf-8")
chunk_data_id = chunk[0:6]