mirror of
https://github.com/affaan-m/ECC.git
synced 2026-08-17 21:15:40 +02:00
1a747097f2
* fix: add .gitattributes to force LF line endings for text files npm run command-registry:check (part of npm test) fails on a fresh clone on Windows with the common core.autocrlf=true setting: git checks out docs/COMMAND-REGISTRY.json with CRLF, but generate-command-registry.js always writes LF, so the strict string comparison in checkRegistry() never matches. Forcing LF via .gitattributes makes checkouts consistent across platforms regardless of a contributor's local autocrlf setting. * fix: normalize CRLF line endings to LF per .gitattributes pyproject.toml, src/llm/__init__.py, src/llm/prompt/builder.py, src/llm/providers/claude.py, and tests/test_builder.py had CRLF line endings committed to the repo, inconsistent with the rest of the codebase. Renormalized via 'git add --renormalize .' now that .gitattributes enforces eol=lf. --------- Co-authored-by: Affaan Mustafa <me@affaanmustafa.com>
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
import pytest
|
|
from llm.core.types import LLMInput, Message, Role, ToolDefinition
|
|
from llm.prompt import PromptBuilder, adapt_messages_for_provider
|
|
from llm.prompt.builder import PromptConfig
|
|
|
|
|
|
class TestPromptBuilder:
|
|
def test_build_without_system(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
builder = PromptBuilder()
|
|
result = builder.build(messages)
|
|
|
|
assert len(result) == 1
|
|
assert result[0].role == Role.USER
|
|
|
|
def test_build_with_system(self):
|
|
messages = [
|
|
Message(role=Role.SYSTEM, content="You are helpful."),
|
|
Message(role=Role.USER, content="Hello"),
|
|
]
|
|
builder = PromptBuilder()
|
|
result = builder.build(messages)
|
|
|
|
assert len(result) == 2
|
|
assert result[0].role == Role.SYSTEM
|
|
|
|
def test_build_adds_system_from_keyword_options(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
builder = PromptBuilder(system_template="You are a pirate.")
|
|
result = builder.build(messages)
|
|
|
|
assert len(result) == 2
|
|
assert "pirate" in result[0].content
|
|
|
|
def test_build_adds_system_from_prompt_config(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
builder = PromptBuilder(config=PromptConfig(system_template="You are a pirate."))
|
|
result = builder.build(messages)
|
|
|
|
assert len(result) == 2
|
|
assert "pirate" in result[0].content
|
|
|
|
def test_rejects_config_with_keyword_options(self):
|
|
with pytest.raises(ValueError, match="Pass either config or PromptBuilder keyword options"):
|
|
PromptBuilder(
|
|
config=PromptConfig(system_template="Configured."),
|
|
system_template="Keyword override.",
|
|
)
|
|
|
|
def test_empty_system_template_does_not_add_blank_system_message(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
builder = PromptBuilder(system_template="")
|
|
result = builder.build(messages)
|
|
|
|
assert result == messages
|
|
|
|
def test_build_with_tools(self):
|
|
messages = [Message(role=Role.USER, content="Search for something")]
|
|
tools = [
|
|
ToolDefinition(name="search", description="Search the web", parameters={}),
|
|
]
|
|
builder = PromptBuilder(include_tools_in_system=True)
|
|
result = builder.build(messages, tools)
|
|
|
|
assert len(result) == 2
|
|
assert "search" in result[0].content
|
|
assert "Available Tools" in result[0].content
|
|
|
|
|
|
class TestAdaptMessagesForProvider:
|
|
def test_adapt_for_claude(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
result = adapt_messages_for_provider(messages, "claude")
|
|
assert len(result) == 1
|
|
|
|
def test_adapt_for_openai(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
result = adapt_messages_for_provider(messages, "openai")
|
|
assert len(result) == 1
|
|
|
|
def test_adapt_for_ollama(self):
|
|
messages = [Message(role=Role.USER, content="Hello")]
|
|
result = adapt_messages_for_provider(messages, "ollama")
|
|
assert len(result) == 1
|