klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Feature] add google ai embedder (#1019)

Browse Source 
Co-authored-by: Deven Patel <deven298@yahoo.com>
This commit is contained in:
Deven Patel
2023-12-18 13:58:01 +05:30
committed by GitHub
parent 6983ebba49
commit c0b5e93967
12 changed files with 115 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/llm/test_base_llm.py
View File

@@ -38,15 +38,9 @@ def test_is_get_llm_model_answer_implemented():
assert llm.get_llm_model_answer() == "Implemented"
def test_stream_query_response(base_llm):
def test_stream_response(base_llm):
answer = ["Chunk1", "Chunk2", "Chunk3"]
result = list(base_llm._stream_query_response(answer))
assert result == answer
def test_stream_chat_response(base_llm):
answer = ["Chunk1", "Chunk2", "Chunk3"]
result = list(base_llm._stream_chat_response(answer))
result = list(base_llm._stream_response(answer))
assert result == answer