mirror of
https://github.com/affaan-m/ECC.git
synced 2026-09-09 19:27:58 +02:00
test: pin Ollama whitespace normalization
This commit is contained in:
@@ -2,7 +2,7 @@ import pytest
|
||||
|
||||
from llm.core.types import Message, Role, ToolDefinition
|
||||
from llm.prompt import PromptBuilder, adapt_messages_for_provider
|
||||
from llm.prompt.builder import PromptConfig
|
||||
from llm.prompt.builder import PromptConfig, get_provider_builder
|
||||
|
||||
|
||||
class TestPromptBuilder:
|
||||
@@ -90,6 +90,7 @@ class TestAdaptMessagesForProvider:
|
||||
|
||||
result = adapt_messages_for_provider(messages, " ollama ", tools)
|
||||
|
||||
assert get_provider_builder(" ollama ").config.tool_format == "text"
|
||||
assert len(result) == 2
|
||||
assert result[0].role == Role.SYSTEM
|
||||
assert "Available Tools" in result[0].content
|
||||
|
||||
Reference in New Issue
Block a user