Exposed additional classes (#18)

* Exposed missing classes * added embedding_model_factory * removed the dependency on numpy. It is now a conditional dependency for local embeddings * Removed image group dependency Messages are not implemented in llmclient anymore --------- Co-authored-by: Mayk Caldas <[email protected]>
Future-House · Dec 9, 2024 · f96865d · f96865d
1 parent 6731118
commit f96865d
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/llmclient/__init__.py b/llmclient/__init__.py
@@ -1,9 +1,16 @@
+from .constants import (
+    CHARACTERS_PER_TOKEN_ASSUMPTION,
+    EXTRA_TOKENS_FROM_USER_ROLE,
+    MODEL_COST_MAP,
+)
 from .embeddings import (
     EmbeddingModel,
     EmbeddingModes,
     HybridEmbeddingModel,
+    LiteLLMEmbeddingModel,
     SentenceTransformerEmbeddingModel,
     SparseEmbeddingModel,
+    embedding_model_factory,
 )
 from .exceptions import (
     JSONSchemaValidationError,
@@ -13,17 +20,28 @@
     LLMModel,
     MultipleCompletionLLMModel,
 )
-from .types import LLMResult
+from .types import (
+    Chunk,
+    Embeddable,
+    LLMResult,
+)
 
 __all__ = [
+    "CHARACTERS_PER_TOKEN_ASSUMPTION",
+    "EXTRA_TOKENS_FROM_USER_ROLE",
+    "MODEL_COST_MAP",
+    "Chunk",
+    "Embeddable",
     "EmbeddingModel",
     "EmbeddingModes",
     "HybridEmbeddingModel",
     "JSONSchemaValidationError",
     "LLMModel",
     "LLMResult",
+    "LiteLLMEmbeddingModel",
     "LiteLLMModel",
     "MultipleCompletionLLMModel",
     "SentenceTransformerEmbeddingModel",
     "SparseEmbeddingModel",
+    "embedding_model_factory",
 ]