vibesurf 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

vibe_surf/backend/shared_state.py

@@ -106,6 +106,7 @@ async def execute_task_background(
         task: str,
         llm_profile_name: str,
         upload_files: Optional[List[str]] = None,
+        agent_mode: str = "thinking",
         db_session=None
 ):
     """Background task execution function for single task with LLM profile support"""
@@ -141,7 +142,8 @@ async def execute_task_background(
         result = await vibesurf_agent.run(
             task=task,
             upload_files=upload_files,
-            session_id=session_id
+            session_id=session_id,
+            agent_mode=agent_mode
         )
 
         # Update task status to completed
@@ -400,8 +402,8 @@ async def initialize_vibesurf_components():
 
         db_manager = DatabaseManager(database_url)
 
-        # Initialize database tables
-        await db_manager.create_tables()
+        # Initialize database tables with migration support
+        await db_manager.create_tables(use_migrations=True)
         logger.info("✅ Database manager initialized successfully")
 
         # Initialize LLM from default profile (if available) or fallback to environment variables

vibe_surf/chrome_extension/scripts/api-client.js

@@ -175,7 +175,8 @@ class VibeSurfAPIClient {
       task_description,
       llm_profile_name,
       upload_files_path,
-      mcp_server_config
+      mcp_server_config,
+      agent_mode
     } = taskData;
 
     return this.post('/tasks/submit', {
@@ -183,7 +184,8 @@ class VibeSurfAPIClient {
       task_description,
       llm_profile_name,
       upload_files_path,
-      mcp_server_config
+      mcp_server_config,
+      agent_mode
     });
   }
 

vibe_surf/chrome_extension/scripts/ui-manager.js

@@ -48,6 +48,7 @@ class VibeSurfUIManager {
       
       // Input area
       llmProfileSelect: document.getElementById('llm-profile-select'),
+      agentModeSelect: document.getElementById('agent-mode-select'),
       taskInput: document.getElementById('task-input'),
       sendBtn: document.getElementById('send-btn'),
       
@@ -236,6 +237,9 @@ class VibeSurfUIManager {
     // LLM profile selection handling
     this.elements.llmProfileSelect?.addEventListener('change', this.handleLlmProfileChange.bind(this));
     
+    // Agent mode selection handling
+    this.elements.agentModeSelect?.addEventListener('change', this.handleAgentModeChange.bind(this));
+    
     // Initialize auto-resize for textarea
     if (this.elements.taskInput) {
       this.autoResizeTextarea(this.elements.taskInput);
@@ -461,6 +465,11 @@ class VibeSurfUIManager {
       this.elements.llmProfileSelect.disabled = isRunning && !isPaused;
     }
     
+    if (this.elements.agentModeSelect) {
+      // Allow agent mode change only when not running
+      this.elements.agentModeSelect.disabled = isRunning && !isPaused;
+    }
+    
     // Update file manager state - keep disabled during pause (as per requirement)
     this.fileManager.setEnabled(!isRunning);
     
@@ -505,6 +514,11 @@ class VibeSurfUIManager {
       this.elements.llmProfileSelect.disabled = true;
     }
     
+    // Keep agent mode disabled during pause (user doesn't need to change it)
+    if (this.elements.agentModeSelect) {
+      this.elements.agentModeSelect.disabled = true;
+    }
+    
     // Keep file manager disabled during pause
     this.fileManager.setEnabled(false);
     
@@ -541,6 +555,10 @@ class VibeSurfUIManager {
       this.elements.llmProfileSelect.disabled = true;
     }
     
+    if (this.elements.agentModeSelect) {
+      this.elements.agentModeSelect.disabled = true;
+    }
+    
     // Update file manager state
     this.fileManager.setEnabled(false);
     
@@ -716,7 +734,8 @@ class VibeSurfUIManager {
       
       const taskData = {
         task_description: taskDescription,
-        llm_profile_name: llmProfile
+        llm_profile_name: llmProfile,
+        agent_mode: this.elements.agentModeSelect?.value || 'thinking'
       };
       
       // Add uploaded files path if any
@@ -867,6 +886,13 @@ class VibeSurfUIManager {
     }
   }
 
+  handleAgentModeChange(event) {
+    // Re-validate send button state when agent mode changes
+    if (this.elements.taskInput) {
+      this.handleTaskInputChange({ target: this.elements.taskInput });
+    }
+  }
+
   handleTaskInputChange(event) {
     console.log('[UIManager] handleTaskInputChange called with value:', event.target.value);
     

vibe_surf/chrome_extension/sidepanel.html

@@ -173,6 +173,11 @@
                 <div class="input-footer">
                     <select id="llm-profile-select" class="llm-select compact">
                     </select>
+                    <select id="agent-mode-select" class="agent-mode-select compact">
+                        <option value="thinking" selected>Thinking</option>
+                        <option value="no-thinking">No-thinking</option>
+                        <option value="flash">Flash</option>
+                    </select>
                 </div>
             </div>
             <input type="file" id="file-input" class="hidden" multiple accept="*/*">

vibe_surf/chrome_extension/styles/input.css

@@ -123,7 +123,8 @@
   margin-top: calc(var(--spacing-xs) / 2);
 }
 
-.llm-select {
+.llm-select,
+.agent-mode-select {
   padding: var(--spacing-sm) var(--spacing-md);
   border: 1px solid var(--border-color);
   border-radius: var(--radius-md);
@@ -136,12 +137,19 @@
 }
 
 .llm-select.compact {
-  max-width: 100px;
+  max-width: 140px;
   font-size: var(--font-size-xs);
   padding: var(--spacing-xs) var(--spacing-sm);
 }
 
-.llm-select:focus {
+.agent-mode-select.compact {
+  max-width: 70px;
+  font-size: var(--font-size-xs);
+  padding: var(--spacing-xs) var(--spacing-sm);
+}
+
+.llm-select:focus,
+.agent-mode-select:focus {
   outline: none;
   border-color: var(--primary-color);
   box-shadow: 0 0 0 3px rgba(0, 122, 204, 0.1);

vibe_surf/llm/openai_compatible.py

@@ -226,7 +226,7 @@ class ChatOpenAICompatible(ChatOpenAI):
         if self._is_qwen_model() or self._is_kimi_model():
             self.add_schema_to_system_prompt = True
 
-        if not (self._is_gemini_model() or self._is_kimi_model()) or output_format is None:
+        if not (self._is_gemini_model() or self._is_kimi_model() or self._is_qwen_model()) or output_format is None:
             return await super().ainvoke(messages, output_format)
         openai_messages = OpenAIMessageSerializer.serialize_messages(messages)
 
@@ -281,7 +281,6 @@ class ChatOpenAICompatible(ChatOpenAI):
                     fixed_schema = self._fix_kimi_schema(original_schema)
                 else:
                     fixed_schema = original_schema
-
                 response_format: JSONSchema = {
                     'name': 'agent_output',
                     'strict': True,

{vibesurf-0.1.15.dist-info → vibesurf-0.1.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibesurf
-Version: 0.1.15
+Version: 0.1.17
 Summary: VibeSurf: A powerful browser assistant for vibe surfing
 Author: Shao Warm
 License: Apache-2.0

{vibesurf-0.1.15.dist-info → vibesurf-0.1.17.dist-info}/RECORD

@@ -1,35 +1,34 @@
 vibe_surf/__init__.py,sha256=WtduuMFGauMD_9dpk4fnRnLTAP6ka9Lfu0feAFNzLfo,339
-vibe_surf/_version.py,sha256=HPqQHR9pVxIxlFt4vovkyoe7k6UO3ag2isBN2lHFL8g,706
+vibe_surf/_version.py,sha256=TtVjjQ5FSnY_MX0ZAPLaNmAfTJWa0sEMBdMs65ngXMM,706
 vibe_surf/cli.py,sha256=pbep2dBeQqralZ8AggkH4h2nayBarbdN8lhZxo35gNU,16689
 vibe_surf/common.py,sha256=_WWMxen5wFwzUjEShn3yDVC1OBFUiJ6Vccadi6tuG6w,1215
 vibe_surf/logger.py,sha256=k53MFA96QX6t9OfcOf1Zws8PP0OOqjVJfhUD3Do9lKw,3043
 vibe_surf/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vibe_surf/agents/browser_use_agent.py,sha256=R8myQ0c9YsJve30yMfIRgz2IzOoHUxrNTtXj8x0YHiw,46032
-vibe_surf/agents/report_writer_agent.py,sha256=tKELps-FjZq06OKXpoyS-zJzvfdYXM6UnXDrR2oZk6M,20970
-vibe_surf/agents/vibe_surf_agent.py,sha256=-pC7MSOd7I21JbLN0yY69RNHRfTFt-STfA4j6jIyrS4,73191
-vibe_surf/agents/views.py,sha256=bcVOxd9On2lJ4NSar5RqP1V7Nw_RBtkHJPE-f2kVdRA,4750
+vibe_surf/agents/browser_use_agent.py,sha256=qmm3TAVzEYjvOvSQEa5xUZOgpBOR_8qePME8BjLjOd0,45951
+vibe_surf/agents/report_writer_agent.py,sha256=0RZHcrk6WoC7GE66QshrPLt6ka7GlRtnCsgjO7Dfxfw,20966
+vibe_surf/agents/vibe_surf_agent.py,sha256=i7O4Djw3Pd1ijwbkGbXkQqO1y0RunlelCqajjfJFfaQ,73687
+vibe_surf/agents/views.py,sha256=yHjNJloa-aofVTGyuRy08tBYP_Y3XLqt1DUWOUmHRng,4825
 vibe_surf/agents/prompts/__init__.py,sha256=l4ieA0D8kLJthyNN85FKLNe4ExBa3stY3l-aImLDRD0,36
 vibe_surf/agents/prompts/report_writer_prompt.py,sha256=sZE8MUT1CDLmRzbnbEQzAvTwJjpITgh2Q8g1_eXmkzE,4454
 vibe_surf/agents/prompts/vibe_surf_prompt.py,sha256=gn8Q234zwtUxVzduzmINmqV-iooE-aJ9u2-yXQ3Wo6M,5573
 vibe_surf/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vibe_surf/backend/llm_config.py,sha256=9V8Gg065TQALbOKQnOqFWd8RzOJjegOD8w6YOf90Q7Y,5036
 vibe_surf/backend/main.py,sha256=nZDs_FkYU7NdqSm1O4OwHJslYIJZL1Gr3CEKa8ByaEY,7178
-vibe_surf/backend/shared_state.py,sha256=yEBWj6YI6WjVF2nuJ_nujo8yqHw2j8xikmxr0G8fnbs,23331
+vibe_surf/backend/shared_state.py,sha256=doZv5A6h2Qvq1hM5a3-vnU0Jc8AAlHwuM9WhVu5mFk4,23446
 vibe_surf/backend/api/__init__.py,sha256=XxF1jUOORpLYCfFuPrrnUGRnOrr6ClH0_MNPU-4RnSs,68
 vibe_surf/backend/api/activity.py,sha256=_cnHusqolt5Hf3KdAf6FK-3sBc-TSaadmb5dJxGI57A,9398
 vibe_surf/backend/api/browser.py,sha256=NXedyZG3NIVRIx5O7d9mHwVWX-Q4_KsX5mSgfKt8UEA,2122
 vibe_surf/backend/api/config.py,sha256=EwzxYvC6HlaVo2OFWjtBmBMjX4eW2q8hp7l2LO2GZV0,27124
 vibe_surf/backend/api/files.py,sha256=kJMG9MWECKXwGh64Q6xvAzNjeZGcLhIEnn65HiMZHKE,11762
-vibe_surf/backend/api/models.py,sha256=iyMBdpDkOfwx0h2yRkjmYjRdFyDErhO8XZcWTurdcmU,10347
-vibe_surf/backend/api/task.py,sha256=jf9y7T5SjgFdw1XxKrHvc7nMpM544-VmUTBBMF4Msgg,14263
+vibe_surf/backend/api/models.py,sha256=n_bu8vavvO8bIKA1WUAbaGPFeZKeamMJelDWU3DlFJc,10533
+vibe_surf/backend/api/task.py,sha256=vpQMOn6YBuD_16jzfUajUvBYaydC0jj8Ny3WOJDVuck,14359
 vibe_surf/backend/database/__init__.py,sha256=XhmcscnhgMhUyXML7m4SnuQIqkFpyY_zJ0D3yYa2RqQ,239
-vibe_surf/backend/database/manager.py,sha256=aC8XQc68qDsaryrux7VfCEl2hnEZL5kfnEkXWXPg2ig,4427
-vibe_surf/backend/database/models.py,sha256=mePuHsaSqJKA4TthvXbup_Ioann2_chxywiLKqAWyh4,7009
-vibe_surf/backend/database/queries.py,sha256=2ff9w7piY6ssIPgqnEM15qCX8gsZuUgmS00fwaZSEHQ,33638
+vibe_surf/backend/database/manager.py,sha256=Okmr6yG2aycmatONRMyRbHe6l53RkFIPeMxxPSD3ycY,11884
+vibe_surf/backend/database/models.py,sha256=AomcjHTcKfJUEI4m7wUobxnruGehwiz6CJJvvDJuQmU,7151
+vibe_surf/backend/database/queries.py,sha256=WEXZBNHQ1LWWR4Bv-49LMs8mZOkapFBx2_vM1GGLoA8,33723
 vibe_surf/backend/database/schemas.py,sha256=OPnpRKwYG1Cu8geJ6pajiEDF8x8mRestXnAfI4Gy18w,3402
-vibe_surf/backend/migrations/__init__.py,sha256=dLhZwW6AGyfBSid-QJPCpIlS4DnYDvO8NyI4s8JAZB8,383
-vibe_surf/backend/migrations/init_db.py,sha256=pY2Yq7K1vPxqT8r3jlAQcYEQWK-GGbb0F7W5asGpuew,10399
-vibe_surf/backend/migrations/seed_data.py,sha256=L6Ll-u8P4cICAUlD5y9urQPSUld6M67erSBCEIdw8Uc,8239
+vibe_surf/backend/database/migrations/v001_initial_schema.sql,sha256=MC2fa1WHUEhHhdOTxz0qB4RI7JdGRpiGXZ77ytl3LRQ,4345
+vibe_surf/backend/database/migrations/v002_add_agent_mode.sql,sha256=jKnW28HsphUeU9kudEx9QaLnUh8swmmOt-hFsZJay24,251
 vibe_surf/backend/utils/__init__.py,sha256=V8leMFp7apAglUAoCHPZrNNcRHthSLYIudIJE5qwjb0,184
 vibe_surf/backend/utils/encryption.py,sha256=CjLNh_n0Luhfa-6BB-icfzkiiDqj5b4Gu6MADU3p2eM,3754
 vibe_surf/backend/utils/llm_factory.py,sha256=mNy8o3sw7vYJ8gwiTsrgXbG7Ri_B11ylE4KGcfHULp8,8972
@@ -47,12 +46,12 @@ vibe_surf/chrome_extension/content.js,sha256=q6JRpmHAEdPWNnFSIqoPv8eBm0T574c3H5h
 vibe_surf/chrome_extension/dev-reload.js,sha256=xQpi-1Ekb5P8Ogsm6rUK09QzxafwH0H409zBKmaUFNw,1790
 vibe_surf/chrome_extension/manifest.json,sha256=B08nHuU-bPc-pUr30Y-of39TjMlrE7D5gP2sZjZ8CrE,1142
 vibe_surf/chrome_extension/popup.html,sha256=n3dI_-WbILm0q8O_za6xX0WvOofz5lwT_7YXs0u9RAE,4248
-vibe_surf/chrome_extension/sidepanel.html,sha256=ams_TCA68787uVRnQ7J70b4rNzMEWzQsgVvSHoabouI,25942
+vibe_surf/chrome_extension/sidepanel.html,sha256=v6FBuKsKIZ5Mvup44rzoJsKV9gBH1ypvr25nu3j2yBk,26268
 vibe_surf/chrome_extension/icons/convert-svg.js,sha256=j137nZA7MJK35NtrwWff8yb3UEKa5XTAvnV6EjY-CVI,953
 vibe_surf/chrome_extension/icons/logo-preview.html,sha256=hrgU45uziKHKIb8be9P4ZrZJyGggWnm2M5oEu89V5sM,6962
 vibe_surf/chrome_extension/icons/logo.icns,sha256=ZzY1eIKF4dNhNW4CeE1UBQloxNVC7bQx3qcClo3CnMQ,1569615
 vibe_surf/chrome_extension/icons/logo.png,sha256=PLmv1E6sCGXUE5ZDxr-pFPQd9Gvaw_f1TnYmF8VIssU,566385
-vibe_surf/chrome_extension/scripts/api-client.js,sha256=YUnMZOKUcJaGFl8wCpGxgCSvBKqbwCtYbQLbI32otrY,13844
+vibe_surf/chrome_extension/scripts/api-client.js,sha256=5H5GU-NsO_V_sEP3JheKJ3ejjxiruKZlM5hoqAaRjFs,13880
 vibe_surf/chrome_extension/scripts/file-manager.js,sha256=WcEaeoECq84jXqEo3Q2ywmqppwSUVm-0dtrbTYXL07g,15809
 vibe_surf/chrome_extension/scripts/history-manager.js,sha256=J_EiBKrRp2cTxuy0Sq5oCju5YuX0_2pGG6NWrJSjC24,22218
 vibe_surf/chrome_extension/scripts/main.js,sha256=WhmCGktQoSl7aaMl8a9ysJJiysAjf12bWGypMucCSVg,16913
@@ -60,13 +59,13 @@ vibe_surf/chrome_extension/scripts/markdown-it.min.js,sha256=gZ3xe0BdCJplNiHWTKr
 vibe_surf/chrome_extension/scripts/modal-manager.js,sha256=9ekFaTmy8sk94KK9HZJ7739ObsZ6ssllmRLP9rQxUKU,14187
 vibe_surf/chrome_extension/scripts/session-manager.js,sha256=sXh4j4dv3eKOGYeov8iUPt-DX5jKQLhcASDXRX9qW5Y,21126
 vibe_surf/chrome_extension/scripts/settings-manager.js,sha256=rgstYooxZtV9QjqP8KRRQclHbeiZJozZZfv80A6M_P4,48601
-vibe_surf/chrome_extension/scripts/ui-manager.js,sha256=_qMrS_3_jZn7lDlXkTlF4CwOZxdg3B3Enww95PaGsHA,81526
+vibe_surf/chrome_extension/scripts/ui-manager.js,sha256=nFwe1vb9S74wD4Z3QreyMn21DuKMBKOMH3BwpvP-jfY,82488
 vibe_surf/chrome_extension/styles/activity.css,sha256=aEFa_abskrDQvwsesPVOyJW3rUQUEBQpMKPHhY94CoA,19601
 vibe_surf/chrome_extension/styles/animations.css,sha256=TLAet_xXBxCA-H36BWP4xBGBIVjbDdAj7Q6OPxPsbE8,7891
 vibe_surf/chrome_extension/styles/base.css,sha256=d8nt7Wej1r57G6pnBIGKEVkepFxlq8Ets0isngCf5w8,1296
 vibe_surf/chrome_extension/styles/components.css,sha256=cbpYgT1ujG11axRa9xnHizgm4wfrpSzq_nkUOdfNZ0A,18495
 vibe_surf/chrome_extension/styles/history-modal.css,sha256=xCg0EPz1N4kg47jnz9q9E1M4zGag8lJMt_msAOkZi3M,16583
-vibe_surf/chrome_extension/styles/input.css,sha256=Sn2QulNV4jS2NJRByp7RVyZtCJbexw48AjLxgpB-XRw,10566
+vibe_surf/chrome_extension/styles/input.css,sha256=7gmWfXbm3CHPEPHtzwSwAtGeNGJtzrect7yYy57rnEo,10745
 vibe_surf/chrome_extension/styles/layout.css,sha256=ggh-lWpP6sHubFkshf9dcmA52LiXw7VJmQXcM5_EGcY,3528
 vibe_surf/chrome_extension/styles/responsive.css,sha256=a_SFX-PeOcenG4xMeoru5XFnnXdf0YhEyRzvTVVZHdI,8529
 vibe_surf/chrome_extension/styles/settings-environment.css,sha256=OzgezmfZZ_yD60Jy9Uiq83mDmMCfk2dPPcZISb9Eqd4,3390
@@ -77,7 +76,7 @@ vibe_surf/chrome_extension/styles/settings-responsive.css,sha256=jLE0yG15n2aI6_6
 vibe_surf/chrome_extension/styles/settings-utilities.css,sha256=3PuQS2857kg83d5erLbLdo_7J95-qV-qyNWS5M-w1oQ,505
 vibe_surf/chrome_extension/styles/variables.css,sha256=VbFJ8L76AaJUqb6gCz9jEG_V6OX4YJa65I2iyeNJIw4,1411
 vibe_surf/llm/__init__.py,sha256=_vDVPo6STf343p1SgMQrF5023hicAx0g83pK2Gbk4Ek,601
-vibe_surf/llm/openai_compatible.py,sha256=rdJTGXW-68yTho1hgzhkjMAocQOK0p_GSl8PmAx_C8Y,14974
+vibe_surf/llm/openai_compatible.py,sha256=8v0LW_-ZoKv4gcO--6_SmU_BLF8XCJaiPxZ6kXFgM4I,14998
 vibe_surf/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vibe_surf/tools/browser_use_tools.py,sha256=gWiBj-YPz7kQYi0yekuvpPQ2Z2-EBLuutF6rf71iOcM,28781
 vibe_surf/tools/file_system.py,sha256=gVdTd6qUfoyVfZMzmX9DXTyBbfXY56jVhUqtfm1qqps,18878
@@ -85,9 +84,9 @@ vibe_surf/tools/mcp_client.py,sha256=OeCoTgyx4MoY7JxXndK6pGHIoyFOhf5r7XCbx25y1Ec
 vibe_surf/tools/report_writer_tools.py,sha256=sUqUFr-_Rs8RJ0Bs77Hrp07kNwRIvHv7ErzSPYSlbTg,705
 vibe_surf/tools/vibesurf_tools.py,sha256=VVhM2IZOXwpGvJ3v9jTXFHmxTxF2jAzhrwqFqxFVuX4,29976
 vibe_surf/tools/views.py,sha256=aPoabrXOCrn5vPCAauMKYvpyP9n0qC2xKn1IbjAGXwE,4218
-vibesurf-0.1.15.dist-info/licenses/LICENSE,sha256=czn6QYya0-jhLnStD9JqnMS-hwP5wRByipkrGTvoXLI,11355
-vibesurf-0.1.15.dist-info/METADATA,sha256=Cy00eGyShMlmGt-aOLG4FBbh-JBmqv1tWBZFdTgUTx8,5313
-vibesurf-0.1.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vibesurf-0.1.15.dist-info/entry_points.txt,sha256=UxqpvMocL-PR33S6vLF2OmXn-kVzM-DneMeZeHcPMM8,48
-vibesurf-0.1.15.dist-info/top_level.txt,sha256=VPZGHqSb6EEqcJ4ZX6bHIuWfon5f6HXl3c7BYpbRqnY,10
-vibesurf-0.1.15.dist-info/RECORD,,
+vibesurf-0.1.17.dist-info/licenses/LICENSE,sha256=czn6QYya0-jhLnStD9JqnMS-hwP5wRByipkrGTvoXLI,11355
+vibesurf-0.1.17.dist-info/METADATA,sha256=G79taMncHYGcyU0aq8EM1UF3EbZy41Lwc9C4G_-0D-U,5313
+vibesurf-0.1.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vibesurf-0.1.17.dist-info/entry_points.txt,sha256=UxqpvMocL-PR33S6vLF2OmXn-kVzM-DneMeZeHcPMM8,48
+vibesurf-0.1.17.dist-info/top_level.txt,sha256=VPZGHqSb6EEqcJ4ZX6bHIuWfon5f6HXl3c7BYpbRqnY,10
+vibesurf-0.1.17.dist-info/RECORD,,

vibe_surf/backend/migrations/__init__.py

@@ -1,16 +0,0 @@
-"""
-Database Migration Scripts for VibeSurf Backend
-
-Contains migration scripts for database schema initialization and updates.
-"""
-
-from .init_db import init_database, create_tables, drop_tables
-from .seed_data import seed_initial_data, create_sample_data
-
-__all__ = [
-    "init_database",
-    "create_tables", 
-    "drop_tables",
-    "seed_initial_data",
-    "create_sample_data"
-]

vibe_surf/backend/migrations/init_db.py

@@ -1,303 +0,0 @@
-"""
-Database Initialization and Migration Scripts - Simplified
-
-Handles database schema creation for Task, LLMProfile, and UploadedFile tables.
-"""
-
-import asyncio
-from typing import Optional
-from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
-from sqlalchemy import text
-from sqlalchemy.exc import SQLAlchemyError
-
-from ..database.models import Base
-
-async def init_database(database_url: str = "sqlite+aiosqlite:///./vibe_surf.db") -> bool:
-    """
-    Initialize the database with Task, LLMProfile, and UploadedFile tables and indexes.
-    
-    Args:
-        database_url: Database connection URL
-        
-    Returns:
-        bool: True if initialization was successful
-    """
-    try:
-        # Create async engine
-        engine = create_async_engine(
-            database_url,
-            echo=False,
-            future=True,
-            pool_pre_ping=True
-        )
-        
-        # Create all tables
-        success = await create_tables(engine)
-        
-        if success:
-            print("✅ Database initialized successfully")
-            print(f"📍 Database URL: {database_url}")
-            
-            # Print table information
-            await print_table_info(engine)
-        else:
-            print("❌ Database initialization failed")
-        
-        await engine.dispose()
-        return success
-        
-    except Exception as e:
-        print(f"❌ Database initialization error: {e}")
-        return False
-
-async def create_tables(engine: AsyncEngine) -> bool:
-    """
-    Create all tables (Task, LLMProfile, UploadedFile) and indexes.
-    
-    Args:
-        engine: SQLAlchemy async engine
-        
-    Returns:
-        bool: True if creation was successful
-    """
-    try:
-        async with engine.begin() as conn:
-            # Create all tables (Task, LLMProfile, UploadedFile)
-            await conn.run_sync(Base.metadata.create_all)
-            
-            # Create additional indexes for performance
-            await create_performance_indexes(conn)
-            
-        print("✅ Database tables created successfully")
-        return True
-        
-    except SQLAlchemyError as e:
-        print(f"❌ Error creating tables: {e}")
-        return False
-
-async def drop_tables(engine: AsyncEngine) -> bool:
-    """
-    Drop all database tables (use with caution).
-    
-    Args:
-        engine: SQLAlchemy async engine
-        
-    Returns:
-        bool: True if drop was successful
-    """
-    try:
-        async with engine.begin() as conn:
-            await conn.run_sync(Base.metadata.drop_all)
-            
-        print("✅ Database tables dropped successfully")
-        return True
-        
-    except SQLAlchemyError as e:
-        print(f"❌ Error dropping tables: {e}")
-        return False
-
-async def create_performance_indexes(conn):
-    """
-    Create additional performance indexes for all tables.
-    
-    Args:
-        conn: Database connection
-    """
-    # Performance indexes for all tables
-    indexes = [
-        # Tasks - for session task tracking
-        "CREATE INDEX IF NOT EXISTS idx_tasks_session_status ON tasks(session_id, status)",
-        "CREATE INDEX IF NOT EXISTS idx_tasks_session_created ON tasks(session_id, created_at DESC)",
-        "CREATE INDEX IF NOT EXISTS idx_tasks_status_created ON tasks(status, created_at DESC)",
-        "CREATE INDEX IF NOT EXISTS idx_tasks_completed ON tasks(completed_at DESC)",
-        
-        # LLMProfile - for profile lookups
-        "CREATE INDEX IF NOT EXISTS idx_llm_profiles_name ON llm_profiles(profile_name)",
-        "CREATE INDEX IF NOT EXISTS idx_llm_profiles_provider ON llm_profiles(provider)",
-        "CREATE INDEX IF NOT EXISTS idx_llm_profiles_created ON llm_profiles(created_at DESC)",
-        
-        # UploadedFile - for file management
-        "CREATE INDEX IF NOT EXISTS idx_uploaded_files_session ON uploaded_files(session_id)",
-        "CREATE INDEX IF NOT EXISTS idx_uploaded_files_session_active ON uploaded_files(session_id, is_deleted)",
-        "CREATE INDEX IF NOT EXISTS idx_uploaded_files_upload_time ON uploaded_files(upload_time DESC)",
-        "CREATE INDEX IF NOT EXISTS idx_uploaded_files_deleted ON uploaded_files(is_deleted, deleted_at)",
-        "CREATE INDEX IF NOT EXISTS idx_uploaded_files_filename ON uploaded_files(original_filename)",
-    ]
-    
-    for index_sql in indexes:
-        try:
-            await conn.execute(text(index_sql))
-            print(f"✅ Created index: {index_sql.split('idx_')[1].split(' ')[0]}")
-        except Exception as e:
-            print(f"⚠️  Index creation warning: {e}")
-
-async def print_table_info(engine: AsyncEngine):
-    """
-    Print information about created tables.
-    
-    Args:
-        engine: SQLAlchemy async engine
-    """
-    try:
-        async with engine.connect() as conn:
-            # Get table list
-            if "sqlite" in str(engine.url):
-                result = await conn.execute(text("SELECT name FROM sqlite_master WHERE type='table'"))
-            else:
-                result = await conn.execute(text("SELECT tablename FROM pg_tables WHERE schemaname='public'"))
-            
-            tables = [row[0] for row in result]
-            
-            print(f"\n📊 Created {len(tables)} tables:")
-            for table in sorted(tables):
-                if not table.startswith('sqlite_'):  # Skip SQLite system tables
-                    print(f"   • {table}")
-            
-            # Get approximate row counts (for SQLite)
-            if "sqlite" in str(engine.url):
-                print(f"\n📈 Table sizes:")
-                for table in sorted(tables):
-                    if not table.startswith('sqlite_'):
-                        try:
-                            count_result = await conn.execute(text(f"SELECT COUNT(*) FROM {table}"))
-                            count = count_result.scalar()
-                            print(f"   • {table}: {count} rows")
-                        except Exception:
-                            print(f"   • {table}: - rows")
-    
-    except Exception as e:
-        print(f"⚠️  Could not retrieve table information: {e}")
-
-async def verify_database_schema(database_url: str = "sqlite+aiosqlite:///./vibe_surf.db") -> bool:
-    """
-    Verify that the database schema is correctly set up.
-    
-    Args:
-        database_url: Database connection URL
-        
-    Returns:
-        bool: True if schema is valid
-    """
-    try:
-        engine = create_async_engine(database_url, echo=False)
-        
-        # Expected tables (Task, LLMProfile, UploadedFile)
-        expected_tables = ['tasks', 'llm_profiles', 'uploaded_files']
-        
-        async with engine.connect() as conn:
-            # Check if all expected tables exist
-            if "sqlite" in str(engine.url):
-                result = await conn.execute(text("SELECT name FROM sqlite_master WHERE type='table'"))
-            else:
-                result = await conn.execute(text("SELECT tablename FROM pg_tables WHERE schemaname='public'"))
-            
-            existing_tables = {row[0] for row in result}
-            missing_tables = set(expected_tables) - existing_tables
-            
-            if missing_tables:
-                print(f"❌ Missing tables: {missing_tables}")
-                await engine.dispose()
-                return False
-            
-            # Test basic operations
-            await conn.execute(text("SELECT 1"))
-            
-        await engine.dispose()
-        print("✅ Database schema verification passed")
-        return True
-        
-    except Exception as e:
-        print(f"❌ Database schema verification failed: {e}")
-        return False
-
-async def reset_database(database_url: str = "sqlite+aiosqlite:///./vibe_surf.db") -> bool:
-    """
-    Reset the database by dropping and recreating all tables.
-    
-    Args:
-        database_url: Database connection URL
-        
-    Returns:
-        bool: True if reset was successful
-    """
-    try:
-        engine = create_async_engine(database_url, echo=False)
-        
-        print("🔄 Resetting database...")
-        
-        # Drop all tables
-        if not await drop_tables(engine):
-            await engine.dispose()
-            return False
-        
-        # Recreate all tables
-        if not await create_tables(engine):
-            await engine.dispose()
-            return False
-        
-        await engine.dispose()
-        print("✅ Database reset completed successfully")
-        return True
-        
-    except Exception as e:
-        print(f"❌ Database reset failed: {e}")
-        return False
-
-async def migrate_database(database_url: str = "sqlite+aiosqlite:///./vibe_surf.db") -> bool:
-    """
-    Apply database migrations and updates.
-    
-    Args:
-        database_url: Database connection URL
-        
-    Returns:
-        bool: True if migration was successful
-    """
-    try:
-        engine = create_async_engine(database_url, echo=False)
-        
-        print("🔄 Running database migrations...")
-        
-        # For now, this is equivalent to create_tables
-        # In the future, this could handle schema updates
-        success = await create_tables(engine)
-        
-        await engine.dispose()
-        
-        if success:
-            print("✅ Database migrations completed successfully")
-        else:
-            print("❌ Database migrations failed")
-            
-        return success
-        
-    except Exception as e:
-        print(f"❌ Database migration failed: {e}")
-        return False
-
-# CLI functions for direct usage
-async def main():
-    """Main function for running migrations from command line."""
-    import sys
-    
-    if len(sys.argv) < 2:
-        print("Usage: python -m backend.migrations.init_db [init|reset|verify|migrate]")
-        return
-    
-    command = sys.argv[1]
-    database_url = sys.argv[2] if len(sys.argv) > 2 else "sqlite+aiosqlite:///./vibe_surf.db"
-    
-    if command == "init":
-        await init_database(database_url)
-    elif command == "reset":
-        await reset_database(database_url)
-    elif command == "verify":
-        await verify_database_schema(database_url)
-    elif command == "migrate":
-        await migrate_database(database_url)
-    else:
-        print(f"Unknown command: {command}")
-        print("Available commands: init, reset, verify, migrate")
-
-if __name__ == "__main__":
-    asyncio.run(main())