hjxdl 0.3.29__py3-none-any.whl → 0.3.31__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.29'
-__version_tuple__ = version_tuple = (0, 3, 29)
+__version__ = version = '0.3.31'
+__version_tuple__ = version_tuple = (0, 3, 31)

hdl/utils/llm/llm_wrapper.py

@@ -56,6 +56,8 @@ class OpenAIWrapper(object):
                 *args,
                 **kwargs
             )
+            if "client_type" not in conf:
+                conf["client_type"] = "chat"
 
     def add_client(
         self,
@@ -96,6 +98,7 @@ class OpenAIWrapper(object):
             host = f"http://{host}:{port}/v1"
         self.client_conf[client_id]['host'] = host
         self.client_conf[client_id]['model'] = model
+        self.client_conf[client_id]['client_type'] = kwargs.pop("client_type", "chat")
         self.client_conf[client_id]['client'] = OpenAI(
             base_url=host,
             api_key=api_key,
@@ -347,3 +350,35 @@ class OpenAIWrapper(object):
                 return
 
         return
+
+    def embedding(
+        self,
+        client_id: str,
+        texts: list[str],
+        model: str = None,
+        **kwargs
+    ):
+        """
+        Generates embeddings for a list of texts using a specified model.
+
+        Args:
+            client_id (str): The ID of the client to use for generating embeddings.
+            texts (list[str]): A list of texts for which to generate embeddings.
+            model (str, optional): The model to use for generating embeddings.
+                If not provided, the model specified in the client configuration will be used.
+            **kwargs: Additional keyword arguments to be passed to the client embedding creation method.
+
+        Returns:
+            list: A list of embeddings corresponding to the input texts.
+        """
+        if not model:
+            model = self.client_conf[client_id]['model']
+
+        client = self.client_conf[client_id]['client']
+        response = client.embeddings.create(
+            input=texts,
+            model=model,
+            **kwargs
+        )
+
+        return [i.embedding for i in response.data]

{hjxdl-0.3.29.dist-info → hjxdl-0.3.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: hjxdl
-Version: 0.3.29
+Version: 0.3.31
 Summary: A collection of functions for Jupyter notebooks
 Home-page: https://github.com/huluxiaohuowa/hdl
 Author: Jianxing Hu

{hjxdl-0.3.29.dist-info → hjxdl-0.3.31.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=f1H6PIKEuqUoOqkVNqRQPR6h3oBTbMF-TCtBkXnI8lc,413
+hdl/_version.py,sha256=bhEYrWy2EPGfW55r0auHlTI13JyXnFgvZ-r2iYCZ3RY,413
 hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
 hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -136,15 +136,15 @@ hdl/utils/llm/chatgr.py,sha256=5F5PJHe8vz3iCfi4TT54DCLRi1UeJshECdVtgvvvao0,3696
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
 hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
-hdl/utils/llm/llm_wrapper.py,sha256=sdf6N19pmwXBCYvHfFv8VVPPBpu5SIyXu21W3bFXSx0,13718
+hdl/utils/llm/llm_wrapper.py,sha256=UkD2MnZGGyI-hns6MyUEBS_Mk-U8s8AJMOUP8E3SZ0c,14988
 hdl/utils/llm/vis.py,sha256=SSP6tOwKLq0hWcpM3twI9TitqzBmKjlcGrnXEWYlCzM,26055
 hdl/utils/llm/visrag.py,sha256=0i-VrxqgiV-J7R3VPshu9oc7-rKjFJOldYik3HDXj6M,10176
 hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
 hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
-hjxdl-0.3.29.dist-info/LICENSE,sha256=lkMiSbeZHBQLB9LJEkS9-L3Z-LBC4yGnKrzHSG8RkPM,2599
-hjxdl-0.3.29.dist-info/METADATA,sha256=BTs6PvUzk9E0AzgN86UFTibWjbUfi0ysphPzzCy__u0,1336
-hjxdl-0.3.29.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-hjxdl-0.3.29.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.3.29.dist-info/RECORD,,
+hjxdl-0.3.31.dist-info/LICENSE,sha256=lkMiSbeZHBQLB9LJEkS9-L3Z-LBC4yGnKrzHSG8RkPM,2599
+hjxdl-0.3.31.dist-info/METADATA,sha256=Uy-RmZJGdOwCbvLsnmO3lU0fRVyUDxX8bm13KrGB3G0,1336
+hjxdl-0.3.31.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+hjxdl-0.3.31.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.3.31.dist-info/RECORD,,