zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore(llm.py): remove unused merge_chat_history function to clean up codebase 

chore(test_fallback_handler.py): remove references to the removed merge_chat_history function in tests to maintain consistency
This commit is contained in:
Ariel Frischer 2025-02-14 14:20:11 -08:00
parent f65918cfd3
commit 69281c31db
2 changed files with 0 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
ra_aid/llm.py
View File

@ -322,22 +322,3 @@ def validate_provider_env(provider: str) -> bool:
if key: if key:
return bool(os.getenv(key)) return bool(os.getenv(key))
return False return False
def merge_chat_history(
original_history: List[BaseMessage], fallback_history: List[BaseMessage]
) -> List[BaseMessage]:
"""Merge original and fallback chat histories while preserving order.
Args:
original_history: The original chat message history
fallback_history: Additional messages from fallback attempts
Returns:
List[BaseMessage]: Combined message history preserving chronological order
Note:
The function appends fallback messages to maintain context for future
interactions while preserving the original conversation flow.
"""
return original_history + fallback_history

6
tests/ra_aid/test_fallback_handler.py
View File

@ -60,19 +60,14 @@ class TestFallbackHandler(unittest.TestCase):
return DummyModel() return DummyModel()
def dummy_merge_chat_history():
return ["merged"]
def dummy_validate_provider_env(provider): def dummy_validate_provider_env(provider):
return True return True
import ra_aid.llm as llm import ra_aid.llm as llm
original_initialize = llm.initialize_llm original_initialize = llm.initialize_llm
original_merge = llm.merge_chat_history
original_validate = llm.validate_provider_env original_validate = llm.validate_provider_env
llm.initialize_llm = dummy_initialize_llm llm.initialize_llm = dummy_initialize_llm
llm.merge_chat_history = dummy_merge_chat_history
llm.validate_provider_env = dummy_validate_provider_env llm.validate_provider_env = dummy_validate_provider_env
self.fallback_handler.tool_failure_consecutive_failures = 2 self.fallback_handler.tool_failure_consecutive_failures = 2
@ -81,7 +76,6 @@ class TestFallbackHandler(unittest.TestCase):
self.assertEqual(self.fallback_handler.tool_failure_consecutive_failures, 0) self.assertEqual(self.fallback_handler.tool_failure_consecutive_failures, 0)
llm.initialize_llm = original_initialize llm.initialize_llm = original_initialize
llm.merge_chat_history = original_merge
llm.validate_provider_env = original_validate llm.validate_provider_env = original_validate
def test_load_fallback_tool_models(self): def test_load_fallback_tool_models(self):