pro-craft 0.1.26__tar.gz → 0.1.27__tar.gz

{pro_craft-0.1.26 → pro_craft-0.1.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pro-craft
-Version: 0.1.26
+Version: 0.1.27
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

{pro_craft-0.1.26 → pro_craft-0.1.27}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pro-craft"
-version = "0.1.26"
+version = "0.1.27"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"

{pro_craft-0.1.26 → pro_craft-0.1.27}/src/pro_craft/prompt_craft/async_.py

@@ -159,7 +159,7 @@ class AsyncIntel():
                                     pool_pre_ping=True,  # 使用前检查连接活性
                                     pool_timeout=30      # 等待连接池中连接的最长时间（秒）
                                            )
-        
+        self.create_specific_database(self.engine,["ai_prompts","ai_usecase"])
 
         if model_name in ["gemini-2.5-flash-preview-05-20-nothinking",]:
             self.llm = BianXieAdapter(model_name = model_name)
@@ -169,8 +169,23 @@ class AsyncIntel():
             print('Use BianXieAdapter')
             self.llm = BianXieAdapter()
             
-    async def create_database(self):
-        async with self.engine.begin() as conn:
+    async def create_specific_database(self,engine, tables_to_create_names: list[str]):
+        async with engine.begin() as conn:
+            # 从 metadata 中获取对应的 Table 对象
+            specific_database_objects = []
+            for table_name in tables_to_create_names:
+                if table_name in PromptBase.metadata.tables:
+                    specific_database_objects.append(PromptBase.metadata.tables[table_name])
+                else:
+                    print(f"Warning: Table '{table_name}' not found in metadata.")
+
+            if specific_database_objects:
+                await conn.run_sync(PromptBase.metadata.create_all, tables=specific_database_objects)
+            else:
+                print("No specific tables to create.")
+
+    async def create_database(self,engine):
+        async with engine.begin() as conn:
             await conn.run_sync(PromptBase.metadata.create_all)
             
     async def _get_latest_prompt_version(self,target_prompt_id,session):
@@ -218,9 +233,7 @@ class AsyncIntel():
 
     async def sync_prompt_data_to_database(self,database_url:str):
         target_engine = create_async_engine(database_url, echo=False)
-        async with target_engine.begin() as conn:
-            await conn.run_sync(PromptBase.metadata.create_all)
-            
+        await self.create_database(target_engine) 
         async with create_async_session(self.engine) as source_session:
             async with create_async_session(target_engine) as target_session:
             

{pro_craft-0.1.26 → pro_craft-0.1.27}/src/pro_craft.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pro-craft
-Version: 0.1.26
+Version: 0.1.27
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown