sopel-ai 1.3.3__py3-none-any.whl → 1.3.5__py3-none-any.whl

sopel_ai/core.py

@@ -3,6 +3,7 @@
 from perplexipy import PERPLEXITY_API_URL
 from perplexipy import PERPLEXITY_DEFAULT_MODEL
 from perplexipy import PerplexityClient
+from perplexipy import __VERSION__ as __P_VERSION__
 from sopel_ai import __VERSION__
 from sopel_ai.errors import SopelAIError
 from tinydb import Query
@@ -14,7 +15,9 @@ from tinydb import TinyDB
 # +++ constants +++
 
 DEFAULT_API_KEY = 'pplx-3a45enterthekeyhere'
-DEFAULT_LLM = PERPLEXITY_DEFAULT_MODEL
+# TODO: Fix this:
+# DEFAULT_LLM = PERPLEXITY_DEFAULT_MODEL.replace('reasoning-', '')
+DEFAULT_LLM = 'sonar-pro'
 DEFAULT_LLM_PROVIDER = 'PerplexityAI'
 DEFAULT_LLM_SERVICE = PERPLEXITY_API_URL
 DEFAULT_LOG_LEVEL = 'info'
@@ -63,8 +66,10 @@ def _checkClientInstance(key: str) -> None:
     global _client
 
     if not _client:
+        # TODO:  fix the default in PerplexiPy at some point.
+        model = PERPLEXITY_DEFAULT_MODEL.replace('-reasoning', '')
         _client = PerplexityClient(key = key, endpoint = PERPLEXITY_API_URL)
-        _client.model = PERPLEXITY_DEFAULT_MODEL
+        _client.model = model
 
 
 def runQuery(query: str, nick: str = None, fileNameDB: str = None, responseLength: int = MAX_RESPONSE_LENGTH, key: str = None) -> str:
@@ -160,7 +165,7 @@ def versionInfo(key: str = None) -> str:
         raise SopelAIError('key argument cannot be empty - set the LLM service API key')
 
     _checkClientInstance(key)
-    return 'sopel_ai v%s using %s' % (__VERSION__, '.'.join([_client.__class__.__module__, _client.__class__.__name__]))
+    return 'sopel_ai v%s using %s version %s' % (__VERSION__, '.'.join([_client.__class__.__module__, _client.__class__.__name__]), __P_VERSION__)
 
 
 def setModelForUser(modelID: int, nick: str, fileNameDB: str, key = None) -> str:
@@ -245,8 +250,11 @@ def getModelForUser(nick: str, fileNameDB: str, key = None) -> str:
     Preference = Query()
     preference = _database.search(Preference.nick == nick)
     if preference:
-        model = preference[0]['model']
         client = PerplexityClient(key = key, endpoint = PERPLEXITY_API_URL)
+        model = preference[0]['model']
+        # TODO: Implement unit test for this case.
+        if model not in client.models.keys():
+            model = tuple(client.models.keys())[0]
         client.model = model
         _clientCache[nick] = client
         return model

{sopel_ai-1.3.3.dist-info → sopel_ai-1.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sopel-ai
-Version: 1.3.3
+Version: 1.3.5
 Summary: Sopel AI - an LLM enhanced chat bot plug-in
 Author-email: The SopelAI team <sopel_ai@cime.net>
 License: BSD-3-Clause
@@ -20,11 +20,11 @@ Classifier: Topic :: Utilities
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
-Requires-Dist: perplexipy (==1.1.3)
+Requires-Dist: perplexipy (==1.2.0)
 Requires-Dist: sopel (>=7.1)
 Requires-Dist: tinydb
 
-% sopel_ai(1) Version 1.3.3 chatbot plugin
+% sopel_ai(1) Version 1.3.5 chatbot plugin
 
 Name
 ====

sopel_ai-1.3.5.dist-info/RECORD

@@ -0,0 +1,11 @@
+sopel_ai/__init__.py,sha256=UXSuXZj_p860rLAiewSe5nbKg-qbIEcXq6jkqYmaMnI,2531
+sopel_ai/config.py,sha256=y0vbvfeDjjfLUOvTJ3W56mAGVpJxeizrXJbCaFW4ZXk,867
+sopel_ai/core.py,sha256=VjugYiKMi_cnirQ_Bj0CWZ3DGMpzDutXU3PspNbcZW4,7685
+sopel_ai/errors.py,sha256=XMVgFk4Rw64Z0UO3ZInp-N6LP0GRG8NFIuXKnhu5rLo,192
+sopel_ai/plugin.py,sha256=51mYp6B_mUtgS6mL-IMhK8Qiz_Rw_qAwmTLdML-qQ2o,6800
+sopel_ai-1.3.5.dist-info/LICENSE.txt,sha256=I8aHapysmbM9F3y-rUfp011GQoosNO5L8pzl7IKgPnE,1531
+sopel_ai-1.3.5.dist-info/METADATA,sha256=xlTVp_OV_pm_LST_oE6OgB9CaaEwROvBfgJjUEgKLaI,6210
+sopel_ai-1.3.5.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+sopel_ai-1.3.5.dist-info/entry_points.txt,sha256=7Juxcn6L4j6F83TjkviiTwiyXLM4gZxAAXFQDR2G_m4,43
+sopel_ai-1.3.5.dist-info/top_level.txt,sha256=kpNMzNEGbhCXkyn7oc3uQPmrX1J6qLxn59IcZBpwSYg,9
+sopel_ai-1.3.5.dist-info/RECORD,,

{sopel_ai-1.3.3.dist-info → sopel_ai-1.3.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: bdist_wheel (0.38.4)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

sopel_ai-1.3.3.dist-info/RECORD

@@ -1,11 +0,0 @@
-sopel_ai/__init__.py,sha256=UXSuXZj_p860rLAiewSe5nbKg-qbIEcXq6jkqYmaMnI,2531
-sopel_ai/config.py,sha256=y0vbvfeDjjfLUOvTJ3W56mAGVpJxeizrXJbCaFW4ZXk,867
-sopel_ai/core.py,sha256=x34P4We66tgo5Q229qvlBvY7VuC2ziq9vXkUULVs06M,7277
-sopel_ai/errors.py,sha256=XMVgFk4Rw64Z0UO3ZInp-N6LP0GRG8NFIuXKnhu5rLo,192
-sopel_ai/plugin.py,sha256=51mYp6B_mUtgS6mL-IMhK8Qiz_Rw_qAwmTLdML-qQ2o,6800
-sopel_ai-1.3.3.dist-info/LICENSE.txt,sha256=I8aHapysmbM9F3y-rUfp011GQoosNO5L8pzl7IKgPnE,1531
-sopel_ai-1.3.3.dist-info/METADATA,sha256=Zkcx8RKiIEprrHnk4OQAVHNqqbUeUPT_6BFEmyVaXus,6210
-sopel_ai-1.3.3.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-sopel_ai-1.3.3.dist-info/entry_points.txt,sha256=7Juxcn6L4j6F83TjkviiTwiyXLM4gZxAAXFQDR2G_m4,43
-sopel_ai-1.3.3.dist-info/top_level.txt,sha256=kpNMzNEGbhCXkyn7oc3uQPmrX1J6qLxn59IcZBpwSYg,9
-sopel_ai-1.3.3.dist-info/RECORD,,