hjxdl 0.1.49__py3-none-any.whl → 0.1.50__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.49'
-__version_tuple__ = version_tuple = (0, 1, 49)
+__version__ = version = '0.1.50'
+__version_tuple__ = version_tuple = (0, 1, 50)

hdl/utils/llm/chat.py

@@ -146,30 +146,42 @@ class OpenAI_M():
     def invoke(
         self,
         prompt : str,
+        image_url: str = None,
         stop: list[str] | None = ["USER:", "ASSISTANT:"],
         # history: list = [],
         model="default_model",
         **kwargs: t.Any,
     ) -> str:
-        """Invoke the chatbot with the given prompt and return the response.
+        """Invoke the chatbot with a prompt and optional image URL.
 
         Args:
             prompt (str): The prompt to provide to the chatbot.
+            image_url (str, optional): The URL of the image to include in the chatbot response. Defaults to None.
             stop (list[str], optional): List of strings that indicate the end of the conversation. Defaults to ["USER:", "ASSISTANT:"].
+            model (str, optional): The model to use for the chatbot. Defaults to "default_model".
             **kwargs: Additional keyword arguments to pass to the chatbot.
 
         Returns:
-            str: The response generated by the chatbot.
+            str: The response message from the chatbot.
         """
+        content = [
+            {"type": "text", "text": prompt},
+        ],
+        if image_url is not None:
+            image_content = {
+                "type": "image_url",
+                "image_url": {
+                    "url": image_url,
+                },
+            }
+            content.append(image_content)
         response = self.client.chat.completions.create(
-            # model=self.model_path,
             messages=[{
                 "role": "user",
-                "content": prompt
+                "content": content
             }],
             stream=False,
             model=model,
-            # model=kwargs.get("model", "default_model")
             **kwargs
         )
         return response.choices[0].message.content
@@ -177,30 +189,42 @@ class OpenAI_M():
     def stream(
         self,
         prompt : str,
+        image_url: str = None,
         stop: list[str] | None = ["USER:", "ASSISTANT:"],
         # history: list = [],
         model="default_model",
         **kwargs: t.Any,
     ):
-        """Generate text completion in a streaming fashion.
+        """Generates text and potentially image completions based on the prompt provided.
 
         Args:
-            prompt (str): The text prompt to generate completion for.
-            stop (list[str], optional): List of strings to stop streaming at. Defaults to ["USER:", "ASSISTANT:"].
-            **kwargs: Additional keyword arguments to pass to the completion API.
+            prompt (str): The text prompt to generate completions from.
+            image_url (str, optional): The URL of the image to be included in the completion. Defaults to None.
+            stop (list[str], optional): List of strings that indicate the end of the conversation. Defaults to ["USER:", "ASSISTANT:"].
+            model (str, optional): The model to use for completion generation. Defaults to "default_model".
+            **kwargs: Additional keyword arguments to pass to the completion generation API.
 
         Yields:
-            str: The generated text completion in a streaming fashion.
+            str: The generated completion content.
         """
+        content = [
+            {"type": "text", "text": prompt},
+        ],
+        if image_url is not None:
+            image_content = {
+                "type": "image_url",
+                "image_url": {
+                    "url": image_url,
+                },
+            }
+            content.append(image_content)
         response = self.client.chat.completions.create(
-            # model=self.model_path,
             messages=[{
                 "role": "user",
-                "content": prompt
+                "content": content
             }],
             stream=True,
             model=model,
-            # model=kwargs.get("model", "default_model")
             **kwargs
         )
 

{hjxdl-0.1.49.dist-info → hjxdl-0.1.50.dist-info}/METADATA

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

{hjxdl-0.1.49.dist-info → hjxdl-0.1.50.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=H7_XEdcrsGpo2K4IO_APhPn7oH870y-Gu-5n41Le4X4,413
+hdl/_version.py,sha256=tDigaWSRikP7GznXAhxQfoonLAxYD4L3aYmQCZ0gu7o,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
@@ -127,7 +127,7 @@ hdl/utils/desc/template.py,sha256=a0UAkkKctt_EHY9UECsIIAwVkGPcM1Hr01HSkRMeIuw,12
 hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/general/glob.py,sha256=8-RCnt6L297wMIfn34ZAMCsGCZUjHG3MGglGZI1cX0g,491
 hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/llm/chat.py,sha256=ZyKcGQtt-pqtXqQ4azSYf90_TlOoyn0glpD-MOvkO54,10922
+hdl/utils/llm/chat.py,sha256=hBSmfV0Y8oewJwV-6pY6UjHoFnrgwnKeVIegxLlA1GY,11909
 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
@@ -135,7 +135,7 @@ hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 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.1.49.dist-info/METADATA,sha256=dBGXFJ9jNmeyPGD3jOc9sZ3Gy0REgT-E9e07Pm5KakA,737
-hjxdl-0.1.49.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-hjxdl-0.1.49.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.49.dist-info/RECORD,,
+hjxdl-0.1.50.dist-info/METADATA,sha256=ZTzzrp0k27gajrobfk2zFr1hdTcBc-4Oy88HY9TZpSQ,737
+hjxdl-0.1.50.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+hjxdl-0.1.50.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.50.dist-info/RECORD,,

{hjxdl-0.1.49.dist-info → hjxdl-0.1.50.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (74.0.0)
+Generator: setuptools (74.1.2)
 Root-Is-Purelib: true
 Tag: py3-none-any