From 71fe56159c8e71bffafd98b59e85534443b9f5df Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 18 Jul 2023 11:03:03 -0400 Subject: [PATCH] clena up Former-commit-id: d44f097563acc29a705ed6888829dad488e96b0f --- .dockerignore => swarms/utils/.dockerignore | 0 swarms/utils/llm.py | 1 + tests/LLM.py | 2 +- 3 files changed, 2 insertions(+), 1 deletion(-) rename .dockerignore => swarms/utils/.dockerignore (100%) 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)