fix statusbar when showing the model in the Copilot

Author: bishoy-at-pieces

lua/pieces/copilot/init.lua

@@ -73,7 +73,15 @@ end
 
 local function update_status_bar()
 	if split ~= nil and type(split.winid) == "number" and vim.api.nvim_win_is_valid(split.winid) == true then
-		vim.api.nvim_win_set_option(split.winid, 'statusline', 'Pieces Model: '.. vim.fn.PiecesGetModel())
+		local success, model_name = pcall(vim.fn.PiecesGetModel)
+		if success and model_name then
+			local status_success, _ = pcall(vim.api.nvim_win_set_option, split.winid, 'statusline', 'Pieces Model: ' .. model_name)
+			if not status_success then
+				print("Failed to set statusline option")
+			end
+		end
+	else
+		print("Failed to get model name")
 	end
 end
 

plugin/pieces.lua

@@ -1,6 +1,6 @@
-require("pieces.copilot")
-require("pieces.assets")
-require("pieces.copilot.slash_commands")
-require("pieces.onboarding")
-require("pieces.feedback")
-require("pieces.tutor")
+require("pieces.copilot")
+require("pieces.assets")
+require("pieces.copilot.slash_commands")
+require("pieces.onboarding")
+require("pieces.feedback")
+require("pieces.tutor")

rplugin/python3/pieces_python/__init__.py

@@ -2,24 +2,31 @@
 import pynvim
 import pip
 
+
 def update_sdks():
-	pip.main(["install","pieces_os_client","--upgrade"])
+    pip.main(["install", "pieces_os_client", "--upgrade"])
+
 
-MIN_SDKS_VERSION = "4.4.0"
+MIN_SDKS_VERSION = "4.4.1"
 try:
-	from pieces_os_client import __version__ as pieces_os_client_version
+    from pieces_os_client import __version__ as pieces_os_client_version
+
+    try:  # If there is any issue in the version checker then it is outdated
+        from pieces_os_client.wrapper.version_compatibility import VersionChecker
 
-	try: # If there is any issue in the version checker then it is outdated
-		from pieces_os_client.wrapper.version_compatibility import VersionChecker
-		VersionChecker.compare("1.0.0","1.0.0") # Check also that the compare is working too
-	except (AttributeError, ModuleNotFoundError):
-		update_sdks()
-		raise ModuleNotFoundError
+        VersionChecker.compare(
+            "1.0.0", "1.0.0"
+        )  # Check also that the compare is working too
+    except (AttributeError, ModuleNotFoundError):
+        update_sdks()
+        raise ModuleNotFoundError
 
-	if VersionChecker.compare(pieces_os_client_version,MIN_SDKS_VERSION) < 0: # We need to be above 4.0.0
-		update_sdks()
-		raise ModuleNotFoundError
-	from .main import Pieces
+    if (
+        VersionChecker.compare(pieces_os_client_version, MIN_SDKS_VERSION) < 0
+    ):  # We need to be above 4.0.0
+        update_sdks()
+        raise ModuleNotFoundError
+    from .main import Pieces
 except ModuleNotFoundError:
-	pip.main(["install","pieces_os_client"])
-	from .main import Pieces
+    pip.main(["install", "pieces_os_client"])
+    from .main import Pieces

rplugin/python3/pieces_python/_version.py

@@ -1 +1 @@
-__version__ = "1.5.1"
+__version__ = "1.5.2"