Merge pull request #110 from haesleinhuepf/bump-default-model-0125

Bump default model 0125

src/bia_bob/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.9.0"
+__version__ = "0.10.0"
 
 __all__ = (
 )

src/bia_bob/_kernel.py

@@ -1,10 +1,10 @@
 from metakernel import MetaKernel
 import sys, os
-
+from ._machinery import DEFAULT_MODEL
 
 class BiABobKernel(MetaKernel):
     implementation = 'BiA-Bob'
-    implementation_version = '1.0'
+    implementation_version = '1.1'
     language = 'prompt'
     language_version = '0.1'
     banner = "BiA-Bob"
@@ -19,13 +19,13 @@ class BiABobKernel(MetaKernel):
         "argv": [sys.executable,
                  "-m", "bia_bob",
                  "-f", "{connection_file}"],
-        "display_name": "BiA-Bob (gpt-4-1106-preview)",
+        "display_name": f"BiA-Bob ({DEFAULT_MODEL})",
         "language": "prompt",
         "codemirror_mode": "prompt",
-        "name": "BiA-Bob-gpt-4-1106-preview",
+        "name": f"BiA-Bob-{DEFAULT_MODEL}",
         "logo": str(os.path.abspath(__file__)) + "/images/logo-64x64.png"
     }
-    model = 'gpt-4-1106-preview'
+    model = DEFAULT_MODEL
     magic_prefixes = dict(magic='%', shell='!', help='?')
     help_suffix = "?"
 

src/bia_bob/_machinery.py

@@ -2,6 +2,8 @@
 from ._utilities import keep_available_packages
 import warnings
 
+DEFAULT_MODEL = 'gpt-4-0125-preview'
+
 class Context:
     variables = None
     model = None
@@ -101,18 +103,7 @@ def combine_user_input(line, cell):
     return user_input
 
 
-class CustomAgent:
-    def __init__(self, model="gpt-3.5-turbo"):
-        self.model = model
-
-    def respond_to_user(self, user_input: str):
-        """Sends a prompt to openAI
-        and shows the text response
-        and pastes the code into the next cell.
-        """
-
-
-def init_assistant(model="gpt-4-1106-preview", auto_execute:bool = False, variables:dict=None):
+def init_assistant(model=DEFAULT_MODEL, auto_execute:bool = False, variables:dict=None):
     """Initialises the assistant.
 
     Parameters

src/bia_bob/_utilities.py

@@ -132,7 +132,10 @@ def create_system_prompt(reusable_variables_block=None):
             bia_bob_plugins = entry_points(group='bia_bob_plugins')
         except TypeError:
             all_plugins = entry_points()
-            bia_bob_plugins = all_plugins['bia_bob_plugins']
+            try:
+                bia_bob_plugins = all_plugins['bia_bob_plugins']
+            except KeyError:
+                bia_bob_plugins = []
 
         additional_instructions = []
         # Iterate over discovered entry points and load them