Skip to content

Commit de99137

Browse files
committed
fixes
1 parent a5f1250 commit de99137

2 files changed

Lines changed: 7 additions & 4 deletions

File tree

tests/cassettes.db

200 KB
Binary file not shown.

tests/langchain/test_factory_function.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,19 @@
22
from uipath_langchain_client.factory import get_chat_model, get_embedding_model
33

44
from tests.langchain.conftest import COMPLETION_MODEL_NAMES, EMBEDDING_MODEL_NAMES
5+
from uipath.llm_client.settings import UiPathBaseSettings
56

67

78
@pytest.mark.vcr
89
class TestFactoryFunction:
910
@pytest.mark.parametrize("model_name", COMPLETION_MODEL_NAMES)
10-
def test_get_chat_model(self, model_name: str):
11-
chat_model = get_chat_model(model_name=model_name)
11+
def test_get_chat_model(self, model_name: str, client_settings: UiPathBaseSettings):
12+
chat_model = get_chat_model(model_name=model_name, client_settings=client_settings)
1213
assert chat_model is not None
1314

1415
@pytest.mark.parametrize("model_name", EMBEDDING_MODEL_NAMES)
15-
def test_get_embedding_model(self, model_name: str):
16-
embedding_model = get_embedding_model(model_name=model_name)
16+
def test_get_embedding_model(self, model_name: str, client_settings: UiPathBaseSettings):
17+
embedding_model = get_embedding_model(
18+
model_name=model_name, client_settings=client_settings
19+
)
1720
assert embedding_model is not None

0 commit comments

Comments
 (0)