Add LLMFunctionWithPrompt to __init__

chain-ml · Sep 19, 2024 · 03a7620 · 03a7620
1 parent 582fdfe
commit 03a7620
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/council/llm/__init__.py b/council/llm/__init__.py
@@ -20,6 +20,7 @@
     ExecuteLLMRequest,
 )
 from .llm_function import LLMFunction, LLMFunctionError, FunctionOutOfRetryError
+from .llm_function_with_prompt import LLMFunctionWithPrompt
 from .monitored_llm import MonitoredLLM
 
 from .chat_gpt_configuration import ChatGPTConfigurationBase

diff --git a/tests/integration/llm/test_llm_function_with_prompt.py b/tests/integration/llm/test_llm_function_with_prompt.py
@@ -3,7 +3,7 @@
 import dotenv
 
 from council import AzureLLM
-from council.llm.llm_function_with_prompt import LLMFunctionWithPrompt
+from council.llm import LLMFunctionWithPrompt
 from council.prompt import LLMPromptConfigObject
 from tests import get_data_filename
 from tests.integration.llm.test_llm_function import SQLResult

diff --git a/tests/unit/llm/test_llm_response_parser_yaml_json.py b/tests/unit/llm/test_llm_response_parser_yaml_json.py
@@ -152,7 +152,7 @@ def test_no_block(self):
 
         assert str(e.exception).strip().endswith("json block is not found")
 
-    def test_incorrect_yaml(self):
+    def test_incorrect_json(self):
         llm = MockLLM.from_response(
             """
 ```json