Skip to content

Commit

Permalink
upstage[patch]: fix standard tests (#10)
Browse files Browse the repository at this point in the history
* property

* update integration tests
  • Loading branch information
ccurme authored Jul 2, 2024
1 parent 540370e commit f7c40bb
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 9 deletions.
7 changes: 4 additions & 3 deletions libs/upstage/poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
Expand Up @@ -2,19 +2,18 @@

from typing import Type

import pytest
from langchain_core.language_models import BaseChatModel
from langchain_standard_tests.integration_tests import ChatModelIntegrationTests

from langchain_upstage import ChatUpstage


class TestUpstageStandard(ChatModelIntegrationTests):
@pytest.fixture
@property
def chat_model_class(self) -> Type[BaseChatModel]:
return ChatUpstage

@pytest.fixture
@property
def chat_model_params(self) -> dict:
return {
"model": "solar-1-mini-chat",
Expand Down
5 changes: 2 additions & 3 deletions libs/upstage/tests/unit_tests/test_chat_models_standard.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,19 +2,18 @@

from typing import Type

import pytest
from langchain_core.language_models import BaseChatModel
from langchain_standard_tests.unit_tests import ChatModelUnitTests

from langchain_upstage import ChatUpstage


class TestUpstageStandard(ChatModelUnitTests):
@pytest.fixture
@property
def chat_model_class(self) -> Type[BaseChatModel]:
return ChatUpstage

@pytest.fixture
@property
def chat_model_params(self) -> dict:
return {
"model": "solar-1-mini-chat",
Expand Down

0 comments on commit f7c40bb

Please sign in to comment.