Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[AUR-361] Setup pre-commit, pytest, GitHub actions, ssh-secret (#3)

Browse Source 
Co-authored-by: trducng <trungduc1992@gmail.com>
This commit is contained in:
ian_Cin
2023-08-30 07:22:01 +07:00
committed by GitHub
parent c3c25db48c
commit 5241edbc46
19 changed files with 268 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_llms_completion_models.py
View File

@@ -1,10 +1,10 @@
from unittest.mock import patch
from langchain.llms import AzureOpenAI as AzureOpenAILC, OpenAI as OpenAILC
from langchain.llms import AzureOpenAI as AzureOpenAILC
from langchain.llms import OpenAI as OpenAILC
from kotaemon.llms.completions.openai import AzureOpenAI, OpenAI
from kotaemon.llms.base import LLMInterface
from kotaemon.llms.completions.openai import AzureOpenAI, OpenAI
_openai_completion_response = {
"id": "cmpl-7qyNoIo6gRSCJR0hi8o3ZKBH4RkJ0",
@@ -41,7 +41,9 @@ def test_azureopenai_model(openai_completion):
openai_completion.assert_called()
output = model("hello world")
assert isinstance(output, LLMInterface), "Output for single text is not LLMInterface"
assert isinstance(
output, LLMInterface
), "Output for single text is not LLMInterface"
@patch(
@@ -67,4 +69,6 @@ def test_openai_model(openai_completion):
openai_completion.assert_called()
output = model("hello world")
assert isinstance(output, LLMInterface), "Output for single text is not LLMInterface"
assert isinstance(
output, LLMInterface
), "Output for single text is not LLMInterface"