vibe-engineering 0.1.7__py3-none-any.whl → 0.1.8__py3-none-any.whl

src/cli/commands.py

@@ -1,4 +1,6 @@
 """CLI commands for vibe-engineering."""
+import json
+
 import typer
 from rich.console import Console
 from rich.table import Table
@@ -8,7 +10,7 @@ from rich.progress import Progress, SpinnerColumn, TextColumn
 from rich import box
 import time
 
-from src.db import MongoDBClient, get_documents
+from src.db import MongoDBClient, get_documents, insert_document
 from src.llm import FireworksClient
 from src.schemas import SpecifySchema
 
@@ -140,84 +142,50 @@ def team():
 
 
 @app.command()
-def specify(
-    prompt: str = typer.Argument(..., help="The specification prompt to process"),
-    output_format: str = typer.Option("json", "--format", "-f", help="Output format: json, yaml, markdown"),
-    save: bool = typer.Option(False, "--save", "-s", help="Save the specification to database"),
-    verbose: bool = typer.Option(False, "--verbose", "-v", help="Show detailed output")
-):
-    """✨ Generate a specification schema using AI."""
+def specify(prompt: str, db_name: str = "master", collection_name: str = "llm"):
+    """Generate a specification schema using LLM and store it in MongoDB."""
     console = Console()
-    
-    console.print(f"\n[bold blue]✨ Generating Specification[/bold blue]\n")
-    
-    if verbose:
-        console.print(f"[cyan]📝 Prompt:[/cyan] {prompt}")
-        console.print(f"[cyan]📄 Format:[/cyan] {output_format}")
-        console.print(f"[cyan]💾 Save:[/cyan] {'Yes' if save else 'No'}")
-        console.print()
-    
-    with Progress(
-        SpinnerColumn(),
-        TextColumn("[progress.description]{task.description}"),
-        console=console
-    ) as progress:
-        task = progress.add_task("🤖 Processing with AI...", total=None)
-        
-        try:
-            llm_client = FireworksClient()
-            response = llm_client.generate_with_schema(
-                prompt=prompt,
-                schema=SpecifySchema.model_json_schema(),
-                schema_name="SpecifySchema",
-            )
-            progress.update(task, completed=True)
 
-            # Display results
-            console.print()
-            result_panel = Panel(
-                response,
-                title="📋 Generated Specification",
-                border_style="green",
-                box=box.ROUNDED
+    try:
+        # Generate schema using LLM
+        llm_client = FireworksClient()
+        response = llm_client.generate_with_schema(
+            prompt=prompt,
+            schema=SpecifySchema.model_json_schema(),
+            schema_name="SpecifySchema",
+        )
+
+        # Parse the JSON response
+        doc = json.loads(response)
+
+        # Store in MongoDB
+        with MongoDBClient() as db_client:
+            inserted_id = insert_document(
+                db_client=db_client,
+                db_name=db_name,
+                collection_name=collection_name,
+                document=doc
             )
-            console.print(result_panel)
-            
-            if save:
-                console.print("\n[yellow]💾 Saving to database...[/yellow]")
-                # TODO: Implement save functionality
-                console.print("[green]✅ Specification saved successfully![/green]")
-            
-            # Next steps
-            next_steps = """[yellow]💡 Next Steps:[/yellow]
-            
-• Review and refine the specification
-• Save to database with [cyan]--save[/cyan] flag
-• Use [cyan]vibe plan[/cyan] to create implementation plan
-• Generate tasks with [cyan]vibe tasks[/cyan]"""
-            
-            steps_panel = Panel(
-                next_steps,
-                title="🚀 What's Next?",
-                border_style="yellow",
-                box=box.ROUNDED
-            )
-            console.print(steps_panel)
-            
-        except Exception as e:
-            progress.update(task, completed=True)
-            console.print(f"\n[red]❌ Error generating specification:[/red] {e}")
-            console.print("[yellow]💡 Check your API keys and network connection[/yellow]")
-
 
-@app.callback()
-def main():
-    """
-    🚀 [bold blue]Vibe Engineering[/bold blue] - AI-powered specification and memory management CLI
-    
-    [dim]Manage your project specifications, team knowledge, and development workflows with AI assistance.[/dim]
-    """
-    pass
+        console.print(f"[green]✓[/green] Document stored in {db_name}.{collection_name}")
+        console.print(f"[dim]Document ID: {inserted_id}[/dim]\n")
+
+        # Create a copy of doc for display, converting ObjectId to string if present
+        display_doc = doc.copy()
+        if "_id" in display_doc:
+            display_doc["_id"] = str(display_doc["_id"])
+
+        # Pretty print the schema with rich
+        console.print("[bold cyan]Generated Schema:[/bold cyan]")
+        from rich.syntax import Syntax
+        json_str = json.dumps(display_doc, indent=2, default=str)
+        syntax = Syntax(json_str, "json", theme="monokai", line_numbers=False)
+        console.print(syntax)
+
+    except json.JSONDecodeError as e:
+        console.print(f"[red]Error parsing JSON response:[/red] {e}")
+    except Exception as e:
+        console.print(f"[red]Error:[/red] {e}")
 
 
 if __name__ == "__main__":

{vibe_engineering-0.1.7.dist-info → vibe_engineering-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibe-engineering
-Version: 0.1.7
+Version: 0.1.8
 Summary: Add your description here
 Home-page: https://github.com/vibeengineering/vibe-engineering
 Author: Vibe Engineering Team

{vibe_engineering-0.1.7.dist-info → vibe_engineering-0.1.8.dist-info}/RECORD

@@ -1,6 +1,6 @@
 src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/cli/__init__.py,sha256=jJ7tHuLwJcPDuk45xRuu8xbMBQL45k5oZ6EcNa9MD8k,94
-src/cli/commands.py,sha256=dlZoLSR9n4InbdeAXJXxKmATMGOwLW70ylXlBIJ_808,7885
+src/cli/commands.py,sha256=9aZFNNwlpAJjqqYv0qo1aYhr86vGZefbB3WrcYovRig,6549
 src/db/__init__.py,sha256=zrfEUYNVRhGAq0DgIWBkxoM8L0jJNgy1_GhtBVJvhYA,470
 src/db/client.py,sha256=a-KZDyJjUWB1y15p1pxPn_q8UPDLVtOSvidx8XvULlU,1741
 src/db/operations.py,sha256=nJ2AJrYOaOi9L8ZJgQjr4eukKbr2al0tFls83S8M8GY,4881
@@ -10,7 +10,7 @@ src/llm/embeddings.py,sha256=wH-aihD6HfkJtUn4TJGpqLz9BYnX5i0SyRT0tf8hb2E,2987
 src/llm/segmentation.py,sha256=zex5Elfk_ph0nFFzszxtagN6o1SmTu1BsHJ9W9Njlb0,4753
 src/schemas/__init__.py,sha256=rdw4M8Gm95XN5sPgfReQO0WLePty-OE-4OYjSFrGlBM,163
 src/schemas/models.py,sha256=aIO8qtG0gdL6pWc0cyhYC1jqevpNIfkOQJYSA3Lc_Ew,1011
-vibe_engineering-0.1.7.dist-info/METADATA,sha256=4qsfe3zn4Ln8TQ3l6z17azo6i9Pn4zW9jrJGbI48n98,7650
-vibe_engineering-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vibe_engineering-0.1.7.dist-info/top_level.txt,sha256=74rtVfumQlgAPzR5_2CgYN24MB0XARCg0t-gzk6gTrM,4
-vibe_engineering-0.1.7.dist-info/RECORD,,
+vibe_engineering-0.1.8.dist-info/METADATA,sha256=t1Yu6zdNmjPriJ20u_VZ3gA-Cy--DMcGIlswCq9Gt_w,7650
+vibe_engineering-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vibe_engineering-0.1.8.dist-info/top_level.txt,sha256=74rtVfumQlgAPzR5_2CgYN24MB0XARCg0t-gzk6gTrM,4
+vibe_engineering-0.1.8.dist-info/RECORD,,