diff --git a/.dockerignore b/swarms/utils/.dockerignore similarity index 100% rename from .dockerignore rename to swarms/utils/.dockerignore diff --git a/swarms/utils/llm.py b/swarms/utils/llm.py index b6d6d0c6..21b9f0d4 100644 --- a/swarms/utils/llm.py +++ b/swarms/utils/llm.py @@ -53,6 +53,7 @@ class LLM: raise # example +from swarms.utils.llm import LLM llm_instance = LLM(openai_api_key="your_openai_key") result = llm_instance.run("Who won the FIFA World Cup in 1998?") print(result) diff --git a/tests/LLM.py b/tests/LLM.py index 9f384ade..57360294 100644 --- a/tests/LLM.py +++ b/tests/LLM.py @@ -3,7 +3,7 @@ import os from unittest.mock import patch, MagicMock from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain -from swarms.utils.LLM import LLM +from swarms.utils.llm import LLM class TestLLM(unittest.TestCase): @patch.object(HuggingFaceHub, '__init__', return_value=None)