adam-community 1.0.8__tar.gz → 1.0.10__tar.gz

{adam_community-1.0.8 → adam_community-1.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: adam_community
-Version: 1.0.8
+Version: 1.0.10
 Summary: Adam Community Tools and Utilities
 Home-page: https://github.com/yourusername/adam-community
 Author: Adam Community

adam_community-1.0.10/adam_community/__version__.py

@@ -0,0 +1 @@
+__version__ = "1.0.10"

{adam_community-1.0.8 → adam_community-1.0.10}/adam_community/cli/build.py

@@ -221,6 +221,16 @@ def create_zip_package(directory: Path) -> str:
                     file_count += 1
                     demos_files += 1
         
+        # 检测并添加图标文件
+        icon_files = []
+        icon_extensions = ['.svg', '.png', '.jpg', '.jpeg']
+        for ext in icon_extensions:
+            icon_file = directory / "config" / f"icon{ext}"
+            if icon_file.exists():
+                zipf.write(icon_file, f"config/icon{ext}")
+                file_count += 1
+                icon_files.append(f"icon{ext}")
+        
         # 根据类型添加不同的文件
         if config_type == "kit":
             other_files = ["long_description.md", "input.json"]
@@ -237,6 +247,8 @@ def create_zip_package(directory: Path) -> str:
     console.print(f"   ✓ 配置文件: 1 个")
     if demos_files > 0:
         console.print(f"   ✓ 演示文件: {demos_files} 个")
+    if icon_files:
+        console.print(f"   ✓ 图标文件: {len(icon_files)} 个 ({', '.join(icon_files)})")
     console.print(f"   ✓ 总计: {file_count} 个文件")
     
     return zip_name

{adam_community-1.0.8 → adam_community-1.0.10}/adam_community/util.py

@@ -19,8 +19,31 @@ ADAM_TASK_ID = os.getenv('ADAM_TASK_ID')
 ADAM_USER_ID = os.getenv('ADAM_USER_ID')
 CONDA_ENV = os.getenv('CONDA_ENV')
 
-# 定义可用的知识库集合
-collections = ["DSDP","MPNN","PySCF","RFdiffusion","gaussian","protenix","GPU4PySCF","MolecularDynamics","RDkit","OpenBabel", "SPONGE", "Xponge", "MDAnalysis", "ABACUS"]
+
+def _build_akb_query_command(query: str, collection: str) -> str:
+    """
+    构建 akb 查询命令，包含目录存在性检查
+    
+    Args:
+        query: 查询字符串
+        collection: 知识库名称
+        
+    Returns:
+        完整的 bash 命令字符串
+    """
+    collection_dir = f"/share/programs/akb/database/{collection}"
+    
+    return f"""if [ ! -d "{collection_dir}" ]; then
+    echo "知识库目录不存在: {collection_dir}"
+    exit 1
+fi
+akb simple-query \\
+    -i {collection_dir}/{collection}.index \\
+    -m /share/programs/BAAI/bge-m3 \\
+    -p {collection_dir}/{collection}.parquet \\
+    -l 5 \\
+    -f json -q "{query}\""""
+
 
 class RAG:
     """
@@ -31,17 +54,9 @@ class RAG:
     """
 
     def call(self, query: str, collection: str):
-
-        if collection not in collections:
-            raise ValueError(f"知识库名称必须是下面之一：{collections}")
-
-        cmd = f"""akb simple-query \
-    -i /share/programs/akb/database/{collection}/{collection}.index \
-    -m /share/programs/BAAI/bge-m3 \
-    -p /share/programs/akb/database/{collection}/{collection}.parquet \
-    -l 5 \
-    -f json -q \"{query}\""""
+        cmd = _build_akb_query_command(query, collection)
         logger.info(f"搜索知识库 {collection}: {query}")
+        
         try:
             result = run(cmd, shell='/bin/bash', check=True, text=True, capture_output=True)
         except CalledProcessError as e:

{adam_community-1.0.8 → adam_community-1.0.10}/adam_community.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: adam_community
-Version: 1.0.8
+Version: 1.0.10
 Summary: Adam Community Tools and Utilities
 Home-page: https://github.com/yourusername/adam-community
 Author: Adam Community

adam_community-1.0.8/adam_community/__version__.py

@@ -1 +0,0 @@
-__version__ = "1.0.8"