Minor fixes (cancelling agent package selectiond and getting a defaul…

…t version if version is not specified).
Sema4AI · Oct 22, 2024 · 25beb43 · 25beb43
1 parent 81b71ad
commit 25beb43
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/sema4ai/src/sema4ai_code/workspace_manager.py b/sema4ai/src/sema4ai_code/workspace_manager.py
@@ -1,8 +1,8 @@
 import os
 from collections import defaultdict
+from collections.abc import Iterator
 from pathlib import Path
 from typing import DefaultDict, Dict, List, Optional, Tuple
-from collections.abc import Iterator
 
 from sema4ai_ls_core.cache import CachedFileInfo
 from sema4ai_ls_core.core_log import get_logger
@@ -137,9 +137,16 @@ def _get_agent_package_organizations(
             for action_package_metadata in self._get_action_package_from_dir(
                 org_directory, curr_cache, new_cache
             ):
+                v = action_package_metadata["yamlContents"].get("version")
+                if v is None:
+                    log.info(
+                        f"No version found for action package: {action_package_metadata.get('name')} in dir: {action_package_metadata.get('directory')}"
+                    )
+                    v = "0.0.1"
+
                 action_packages[action_package_metadata["name"]].append(
                     (
-                        str(action_package_metadata["yamlContents"]["version"]),
+                        v,
                         action_package_metadata,
                     )
                 )

diff --git a/sema4ai/vscode-client/src/robo/agentPackage.ts b/sema4ai/vscode-client/src/robo/agentPackage.ts
@@ -264,6 +264,10 @@ export const refreshAgentSpec = async (agentPath: string): Promise<void> => {
 
     if (!agentPath) {
         agentPath = await selectAgentPackage();
+        if (!agentPath) {
+            // Operation cancelled.
+            return;
+        }
     }
 
     try {