lollms-client 0.27.3__py3-none-any.whl → 0.28.0__py3-none-any.whl

lollms_client/__init__.py

@@ -8,7 +8,7 @@ from lollms_client.lollms_utilities import PromptReshaper # Keep general utiliti
 from lollms_client.lollms_mcp_binding import LollmsMCPBinding, LollmsMCPBindingManager
 from lollms_client.lollms_llm_binding import LollmsLLMBindingManager
 
-__version__ = "0.27.3" # Updated version
+__version__ = "0.28.0" # Updated version
 
 # Optionally, you could define __all__ if you want to be explicit about exports
 __all__ = [

lollms_client/llm_bindings/lollms_chat/__init__.py

@@ -442,16 +442,40 @@ class LollmsChatBinding(LollmsLLMBinding):
         
     def embed(self, text: str, **kwargs) -> list:
         """
-        Get embeddings for the input text using OpenAI API
-        
+        Get embeddings for the input text using OpenAI API.
+
         Args:
-            text (str or List[str]): Input text to embed
-            **kwargs: Additional arguments like model, truncate, options, keep_alive
-        
+            text (str): Input text to embed.
+            **kwargs: Additional arguments. The 'model' argument can be used 
+                      to specify the embedding model (e.g., "text-embedding-3-small").
+                      Defaults to "text-embedding-ada-002".
+
         Returns:
-            dict: Response containing embeddings
+            list: The embedding vector as a list of floats, or an empty list on failure.
         """
-        pass    
+        # Determine the embedding model, prioritizing kwargs, with a default
+        embedding_model = kwargs.get("model", self.model_name)
+        
+        try:
+            # The OpenAI API expects the input to be a list of strings
+            response = self.client.embeddings.create(
+                model=embedding_model,
+                input=[text]  # Wrap the single text string in a list
+            )
+            
+            # Extract the embedding from the response
+            if response.data and len(response.data) > 0:
+                return response.data[0].embedding
+            else:
+                ASCIIColors.warning("OpenAI API returned no data for the embedding request.")
+                return []
+                
+        except Exception as e:
+            ASCIIColors.error(f"Failed to generate embeddings using OpenAI API: {e}")
+            trace_exception(e)
+            return []
+        
+
     def get_model_info(self) -> dict:
         """
         Return information about the current OpenAI model.

lollms_client/llm_bindings/openai/__init__.py

@@ -472,16 +472,39 @@ class OpenAIBinding(LollmsLLMBinding):
         
     def embed(self, text: str, **kwargs) -> list:
         """
-        Get embeddings for the input text using OpenAI API
-        
+        Get embeddings for the input text using OpenAI API.
+
         Args:
-            text (str or List[str]): Input text to embed
-            **kwargs: Additional arguments like model, truncate, options, keep_alive
-        
+            text (str): Input text to embed.
+            **kwargs: Additional arguments. The 'model' argument can be used 
+                      to specify the embedding model (e.g., "text-embedding-3-small").
+                      Defaults to "text-embedding-ada-002".
+
         Returns:
-            dict: Response containing embeddings
+            list: The embedding vector as a list of floats, or an empty list on failure.
         """
-        pass    
+        # Determine the embedding model, prioritizing kwargs, with a default
+        embedding_model = kwargs.get("model", self.model_name)
+        
+        try:
+            # The OpenAI API expects the input to be a list of strings
+            response = self.client.embeddings.create(
+                model=embedding_model,
+                input=[text]  # Wrap the single text string in a list
+            )
+            
+            # Extract the embedding from the response
+            if response.data and len(response.data) > 0:
+                return response.data[0].embedding
+            else:
+                ASCIIColors.warning("OpenAI API returned no data for the embedding request.")
+                return []
+                
+        except Exception as e:
+            ASCIIColors.error(f"Failed to generate embeddings using OpenAI API: {e}")
+            trace_exception(e)
+            return []
+
     def get_model_info(self) -> dict:
         """
         Return information about the current OpenAI model.
@@ -574,4 +597,4 @@ class OpenAIBinding(LollmsLLMBinding):
         """
         self.model = model_name
         self.model_name = model_name
-        return True
+        return True

{lollms_client-0.27.3.dist-info → lollms_client-0.28.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lollms_client
-Version: 0.27.3
+Version: 0.28.0
 Summary: A client library for LoLLMs generate endpoint
 Author-email: ParisNeo <parisneoai@gmail.com>
 License: Apache Software License

{lollms_client-0.27.3.dist-info → lollms_client-0.28.0.dist-info}/RECORD

@@ -29,7 +29,7 @@ examples/mcp_examples/openai_mcp.py,sha256=7IEnPGPXZgYZyiES_VaUbQ6viQjenpcUxGiHE
 examples/mcp_examples/run_remote_mcp_example_v2.py,sha256=bbNn93NO_lKcFzfIsdvJJijGx2ePFTYfknofqZxMuRM,14626
 examples/mcp_examples/run_standard_mcp_example.py,sha256=GSZpaACPf3mDPsjA8esBQVUsIi7owI39ca5avsmvCxA,9419
 examples/test_local_models/local_chat.py,sha256=slakja2zaHOEAUsn2tn_VmI4kLx6luLBrPqAeaNsix8,456
-lollms_client/__init__.py,sha256=KvZT0D6ahj16AharXFGhRMw2qc4a4HXDLYO75H5medY,1147
+lollms_client/__init__.py,sha256=eAfY5Ik399J8AVHNU6vLjqiWNNmT88GX-h3NTKzLo2g,1147
 lollms_client/lollms_config.py,sha256=goEseDwDxYJf3WkYJ4IrLXwg3Tfw73CXV2Avg45M_hE,21876
 lollms_client/lollms_core.py,sha256=QliAE6hYiIGqQSDWWmjii1Hj8hsj3IxwBuUFshxJDlM,169099
 lollms_client/lollms_discussion.py,sha256=tvANNvpTkUr4L6GKowosIyfV7l3SA6cXnzElt36e2s8,52133
@@ -55,11 +55,11 @@ lollms_client/llm_bindings/hugging_face_inference_api/__init__.py,sha256=PxgeRqT
 lollms_client/llm_bindings/litellm/__init__.py,sha256=pNkwyRPeENvTM4CDh6Pj3kQfxHfhX2pvXhGJDjKjp30,12340
 lollms_client/llm_bindings/llamacpp/__init__.py,sha256=Qj5RvsgPeHGNfb5AEwZSzFwAp4BOWjyxmm9qBNtstrc,63716
 lollms_client/llm_bindings/lollms/__init__.py,sha256=7GNv-YyX4YyaR1EP2banQEHnX47QpUyNZU6toAiX1ak,17854
-lollms_client/llm_bindings/lollms_chat/__init__.py,sha256=UU3Ep1YkB_pAPkZEA9PpuY_f2KUXghVyzAnUD-u2-tg,23035
+lollms_client/llm_bindings/lollms_chat/__init__.py,sha256=zCCrwAW-lOjAi7nUbvHii7E0kqJiMC0v4ENdbxTUsOs,24114
 lollms_client/llm_bindings/mistral/__init__.py,sha256=624Gr462yBh52ttHFOapKgJOn8zZ1vZcTEcC3i4FYt8,12750
 lollms_client/llm_bindings/ollama/__init__.py,sha256=1yxw_JXye_8l1YaEznK5QhOZmLV_opY-FkYnwy530eo,36109
 lollms_client/llm_bindings/open_router/__init__.py,sha256=v91BpNcuQCbbA6r82gbgMP8UYhSrJUMOf4UtOzEo18Q,13235
-lollms_client/llm_bindings/openai/__init__.py,sha256=pXuLDI7Js0LLXVDCHIMMB5rAQb_wC6JqGvK8B5S0Uy0,24596
+lollms_client/llm_bindings/openai/__init__.py,sha256=kLG0-FyLMoSbEay1hcK46XjEBaLbFE3U51lUjAZ8HoI,25663
 lollms_client/llm_bindings/openllm/__init__.py,sha256=xv2XDhJNCYe6NPnWBboDs24AQ1VJBOzsTuMcmuQ6xYY,29864
 lollms_client/llm_bindings/pythonllamacpp/__init__.py,sha256=7dM42TCGKh0eV0njNL1tc9cInhyvBRIXzN3dcy12Gl0,33551
 lollms_client/llm_bindings/tensor_rt/__init__.py,sha256=nPaNhGRd-bsG0UlYwcEqjd_UagCMEf5VEbBUW-GWu6A,32203
@@ -92,8 +92,8 @@ lollms_client/tts_bindings/piper_tts/__init__.py,sha256=0IEWG4zH3_sOkSb9WbZzkeV5
 lollms_client/tts_bindings/xtts/__init__.py,sha256=FgcdUH06X6ZR806WQe5ixaYx0QoxtAcOgYo87a2qxYc,18266
 lollms_client/ttv_bindings/__init__.py,sha256=UZ8o2izQOJLQgtZ1D1cXoNST7rzqW22rL2Vufc7ddRc,3141
 lollms_client/ttv_bindings/lollms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client-0.27.3.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-lollms_client-0.27.3.dist-info/METADATA,sha256=mkv0Y0vIE5s2o5TwsgZmODaY7e4crlfr1G5OWwMCD8Q,25778
-lollms_client-0.27.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lollms_client-0.27.3.dist-info/top_level.txt,sha256=NI_W8S4OYZvJjb0QWMZMSIpOrYzpqwPGYaklhyWKH2w,23
-lollms_client-0.27.3.dist-info/RECORD,,
+lollms_client-0.28.0.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+lollms_client-0.28.0.dist-info/METADATA,sha256=Esa4mMjN3hLYTMUR8aGk9jNlE_kqCyYLLuWx-tL9Vzo,25778
+lollms_client-0.28.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lollms_client-0.28.0.dist-info/top_level.txt,sha256=NI_W8S4OYZvJjb0QWMZMSIpOrYzpqwPGYaklhyWKH2w,23
+lollms_client-0.28.0.dist-info/RECORD,,