hjxdl 0.3.3__py3-none-any.whl → 0.3.5__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.3'
-__version_tuple__ = version_tuple = (0, 3, 3)
+__version__ = version = '0.3.5'
+__version_tuple__ = version_tuple = (0, 3, 5)

hdl/utils/llm/chat.py

@@ -127,11 +127,10 @@ class OpenAI_M():
         model_path: str = "default_model",
         device: str='gpu',
         generation_kwargs: dict = None,
+        server_host: str = None,
         server_ip: str = "172.28.1.2",
         server_port: int = 8000,
         api_key: str = "dummy_key",
-        use_groq: bool = False,
-        groq_api_key: str = None,
         tools: list = None,
         tool_desc: dict = None,
         cot_desc: str = None,
@@ -149,8 +148,6 @@ class OpenAI_M():
             server_ip (str): IP address of the server. Defaults to "172.28.1.2".
             server_port (int): Port number of the server. Defaults to 8000.
             api_key (str): API key for authentication. Defaults to "dummy_key".
-            use_groq (bool): Flag to use Groq client. Defaults to False.
-            groq_api_key (str, optional): API key for Groq client.
             tools (list, optional): List of tools to be used.
             tool_desc (dict, optional): Additional tool descriptions.
             cot_desc (str, optional): Chain of Thought description.
@@ -160,21 +157,19 @@ class OpenAI_M():
         # self.model_path = model_path
         self.server_ip = server_ip
         self.server_port = server_port
-        self.base_url = f"http://{self.server_ip}:{str(self.server_port)}/v1"
+        self.server_host = server_host
+        if self.server_ip:
+            self.base_url = f"http://{self.server_ip}:{str(self.server_port)}/v1"
+        elif self.server_host:
+            self.base_url = self.host
         self.api_key = api_key
-        self.use_groq = use_groq
-        if use_groq:
-            import groq
-            self.client = groq.Groq(
-                api_key=os.getenv("GROQ_API_KEY", groq_api_key)
-            )
-        else:
-            self.client = OpenAI(
-                base_url=self.base_url,
-                api_key=self.api_key,
-                *args,
-                **kwargs
-            )
+
+        self.client = OpenAI(
+            base_url=self.base_url,
+            api_key=self.api_key,
+            *args,
+            **kwargs
+        )
         self.tools: list = tools if tools else []
         self.tool_desc: dict = TOOL_DESC
         if tool_desc is not None:

{hjxdl-0.3.3.dist-info → hjxdl-0.3.5.dist-info}/METADATA

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

{hjxdl-0.3.3.dist-info → hjxdl-0.3.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=FKnJIExgNrZG2xJ0y_dGNBpxGbGBYylvfat-jHhLUuM,411
+hdl/_version.py,sha256=3eLsZPTwWh0zKBhR6n3c4iAL9geCCdJGMojIL6dF0IA,411
 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
@@ -128,7 +128,7 @@ hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 hdl/utils/general/glob.py,sha256=Zuf7WHU0UdUPOs9UrhxmrCiMC8GrHxQU6n3mTThv6yc,1120
 hdl/utils/general/runners.py,sha256=x7QBolp3MrqNV6L4rB6Ueybr26bqkRFZTuXhY0SwyLk,3061
 hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/llm/chat.py,sha256=SHZdmdLwFgyy7_jnyzsfGIJ0vnGBY4jMm9R1OKjgk88,26280
+hdl/utils/llm/chat.py,sha256=9rrFfmZhiJYIQXsGV78k1L0EH5ohrBxI_xRxZO5ld-c,26016
 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
@@ -139,7 +139,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.3.3.dist-info/METADATA,sha256=RmV3TixHV087g0tdQI0hrj3o7Qzpk5c1q36YTLVk25I,861
-hjxdl-0.3.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-hjxdl-0.3.3.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.3.3.dist-info/RECORD,,
+hjxdl-0.3.5.dist-info/METADATA,sha256=-fAX1nKcpJG37qxIIKPR2RTj3HXS0cEBJVY-jpaN4G0,861
+hjxdl-0.3.5.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+hjxdl-0.3.5.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.3.5.dist-info/RECORD,,

{hjxdl-0.3.3.dist-info → hjxdl-0.3.5.dist-info}/WHEEL

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