External/ECC
1
0
Fork 0
mirror of https://github.com/affaan-m/ECC.git synced 2026-09-08 07:37:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

Trim provider names in prompt builder

Browse Source
This commit is contained in:
Aditya Datta
2026-08-24 22:27:42 -03:00
committed by Alex Schmitt
parent ea2ec0d249
commit 9d233aaa63
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/llm/prompt/builder.py
+1 -1
View File
@@ -118,7 +118,7 @@ _PROVIDER_TEMPLATE_MAP: dict[str, dict[str, Any]] = {
def get_provider_builder(provider_name: str) -> PromptBuilder:
config_dict = _PROVIDER_TEMPLATE_MAP.get(provider_name.lower(), {})
config_dict = _PROVIDER_TEMPLATE_MAP.get(provider_name.strip().lower(), {})
config = PromptConfig(**config_dict)
return PromptBuilder(config)
tests/test_builder.py
+10
View File
@@ -83,3 +83,13 @@ class TestAdaptMessagesForProvider:
messages = [Message(role=Role.USER, content="Hello")]
result = adapt_messages_for_provider(messages, "ollama")
assert len(result) == 1
def test_provider_names_allow_outer_whitespace(self):
messages = [Message(role=Role.USER, content="Hello")]
tools = [ToolDefinition(name="search", description="Search the web", parameters={})]
result = adapt_messages_for_provider(messages, " ollama ", tools)
assert len(result) == 2
assert result[0].role == Role.SYSTEM
assert "Available Tools" in result[0].content