From 848166ef944120860606a7133a603b5ff2b135c2 Mon Sep 17 00:00:00 2001
From: Mayk Caldas <maykcaldas@gmail.com>
Date: Thu, 5 Dec 2024 17:54:02 -0800
Subject: [PATCH] Enabled import directly from llmclient (#14)

Co-authored-by: Mayk Caldas <mayk@futurehouse.org>
---
 llmclient/__init__.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 llmclient/llms.py     |  4 ++--
 2 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/llmclient/__init__.py b/llmclient/__init__.py
index e69de29..985a4c4 100644
--- a/llmclient/__init__.py
+++ b/llmclient/__init__.py
@@ -0,0 +1,43 @@
+from .embeddings import (
+    EmbeddingModel,
+    EmbeddingModes,
+    HybridEmbeddingModel,
+    SentenceTransformerEmbeddingModel,
+    SparseEmbeddingModel,
+)
+from .exceptions import (
+    JSONSchemaValidationError,
+    MalformedMessageError,
+)
+from .llms import (
+    LiteLLMModel,
+    LLMModel,
+    MultipleCompletionLLMModel,
+)
+from .messages import (
+    Message,
+)
+from .types import LLMResult
+from .utils import (
+    encode_image_to_base64,
+    is_coroutine_callable,
+    setup_default_logs,
+)
+
+__all__ = [
+    "EmbeddingModel",
+    "EmbeddingModes",
+    "HybridEmbeddingModel",
+    "JSONSchemaValidationError",
+    "LLMModel",
+    "LLMResult",
+    "LiteLLMModel",
+    "MalformedMessageError",
+    "Message",
+    "MultipleCompletionLLMModel",
+    "SentenceTransformerEmbeddingModel",
+    "SparseEmbeddingModel",
+    "encode_image_to_base64",
+    "is_coroutine_callable",
+    "setup_default_logs",
+]
diff --git a/llmclient/llms.py b/llmclient/llms.py
index 1066e08..d2f8d40 100644
--- a/llmclient/llms.py
+++ b/llmclient/llms.py
@@ -605,8 +605,8 @@ class MultipleCompletionLLMModel(BaseModel):
         description=(
             "Configuration of the model:"
             "model is the name of the llm model to use,"
-            "temperature is the sampling temperature, and",
-            "n is the number of completions to generate.",
+            "temperature is the sampling temperature, and"
+            "n is the number of completions to generate."
         ),
     )
     encoding: Any | None = None