Enabled import directly from llmclient (#14)

Co-authored-by: Mayk Caldas <[email protected]>

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",
+]

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