Backport PR jupyterlab#665: %ai version added

packages/jupyter-ai-magics/jupyter_ai_magics/magics.py

@@ -16,6 +16,7 @@
 from langchain.chains import LLMChain
 from langchain.schema import HumanMessage
 
+from ._version import __version__
 from .parsers import (
     CellArgs,
     DeleteArgs,
@@ -24,6 +25,7 @@
     ListArgs,
     RegisterArgs,
     UpdateArgs,
+    VersionArgs,
     cell_magic_parser,
     line_magic_parser,
 )
@@ -486,6 +488,9 @@ def handle_list(self, args: ListArgs):
             self._ai_list_command_markdown(args.provider_id),
         )
 
+    def handle_version(self, args: VersionArgs):
+        return __version__
+
     def run_ai_cell(self, args: CellArgs, prompt: str):
         # Apply a prompt template.
         prompt = PROMPT_TEMPLATES_BY_FORMAT[args.format].format(prompt=prompt)
@@ -619,6 +624,8 @@ def ai(self, line, cell=None):
                 return self.handle_delete(args)
             if args.type == "update":
                 return self.handle_update(args)
+            if args.type == "version":
+                return self.handle_version(args)
         except ValueError as e:
             print(e, file=sys.stderr)
             return

packages/jupyter-ai-magics/jupyter_ai_magics/parsers.py

@@ -69,6 +69,10 @@ class HelpArgs(BaseModel):
     type: Literal["help"] = "help"
 
 
+class VersionArgs(BaseModel):
+    type: Literal["version"] = "version"
+
+
 class ListArgs(BaseModel):
     type: Literal["list"] = "list"
     provider_id: Optional[str]
@@ -215,6 +219,14 @@ def error_subparser(**kwargs):
     return ErrorArgs(**kwargs)
 
 
+@line_magic_parser.command(name="version")
+def version_subparser(**kwargs):
+    """
+    Prints Jupyter-AI version
+    """
+    return VersionArgs(**kwargs)
+
+
 @line_magic_parser.command(name="help")
 def help_subparser():
     """Show this message and exit."""