mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-13 06:40:04 +00:00
18 lines
428 B
Python
18 lines
428 B
Python
"""Test chat model integration."""
|
|
|
|
from typing import Dict, Type
|
|
|
|
from langchain_standard_tests.unit_tests import ChatModelUnitTests
|
|
|
|
from langchain_ollama.chat_models import ChatOllama
|
|
|
|
|
|
class TestChatOllama(ChatModelUnitTests):
|
|
@property
|
|
def chat_model_class(self) -> Type[ChatOllama]:
|
|
return ChatOllama
|
|
|
|
@property
|
|
def chat_model_params(self) -> Dict:
|
|
return {"model": "llama3-groq-tool-use"}
|