Update tests for vector_store_code
This commit is contained in:
parent
24bfef99a2
commit
0bff803b91
7 changed files with 210 additions and 10 deletions
|
@ -4,18 +4,10 @@ Unit tests for llm chat client functionality
|
|||
|
||||
import pytest
|
||||
|
||||
from reviewllama.configs import create_ollama_config
|
||||
from reviewllama.llm import chat_with_client, create_chat_client
|
||||
from reviewllama.utilities import is_ollama_available
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def ollama_config():
|
||||
return create_ollama_config(
|
||||
"gemma3:4b", "localhost:11434", "You are a helpful assistant.", 0.0
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def chat_client(ollama_config):
|
||||
return create_chat_client(ollama_config)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue