From 92911d728e4c03d75262f5521226bd17ecf78324 Mon Sep 17 00:00:00 2001
From: Luke Hinds <luke@stacklok.com>
Date: Wed, 19 Feb 2025 10:28:23 +0000
Subject: [PATCH] Remove lower case strip
---
src/mockllm/config.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/mockllm/config.py b/src/mockllm/config.py
index 9f58895..5809a3f 100644
--- a/src/mockllm/config.py
+++ b/src/mockllm/config.py
@@ -53,7 +53,7 @@ def load_responses(self) -> None:
def get_response(self, prompt: str) -> str:
"""Get response for a given prompt."""
self.load_responses() # Check for updates
- return self.responses.get(prompt.lower().strip(), self.default_response)
+ return self.responses.get(prompt , self.default_response)
def get_streaming_response(
self, prompt: str, chunk_size: Optional[int] = None