cognify-code 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

ai_code_assistant/cli.py

@@ -156,41 +156,91 @@ def review(ctx, files: Tuple[Path, ...], review_type: str, output_format: str,
               type=click.Choice(["console", "markdown", "json"]))
 @click.option("--source", "-s", type=click.Path(exists=True, path_type=Path),
               help="Source file (for test mode)")
+@click.option("--stream/--no-stream", default=True, help="Stream output in real-time")
 @click.pass_context
 def generate(ctx, description: str, mode: str, language: str, name: Optional[str],
              params: Optional[str], output: Optional[Path], output_format: str,
-             source: Optional[Path]):
+             source: Optional[Path], stream: bool):
     """Generate code from natural language description."""
+    from rich.live import Live
+    from rich.markdown import Markdown
+    from rich.panel import Panel
+    
     config, llm = get_components(ctx.obj.get("config_path"))
     generator = CodeGenerator(config, llm)
     formatter = get_formatter(output_format, config.output.use_colors)
 
     console.print(f"\n[bold]Generating {mode} in {language}...[/bold]\n")
 
-    with console.status("[bold green]Generating code..."):
-        if mode == "function":
-            result = generator.generate_function(
-                description=description, name=name or "generated_function",
-                language=language, parameters=params or "",
-            )
-        elif mode == "class":
-            result = generator.generate_class(
-                description=description, name=name or "GeneratedClass", language=language,
-            )
-        elif mode == "script":
-            result = generator.generate_script(
-                description=description, requirements=[description], language=language,
-            )
-        elif mode == "test":
-            if not source:
-                console.print("[red]Error:[/red] --source required for test mode")
-                sys.exit(1)
-            source_code = source.read_text()
-            result = generator.generate_tests(source_code=source_code, language=language)
-        else:
-            result = generator.generate(description=description, language=language)
+    # Handle test mode source requirement
+    source_code = ""
+    if mode == "test":
+        if not source:
+            console.print("[red]Error:[/red] --source required for test mode")
+            sys.exit(1)
+        source_code = source.read_text()
+
+    if stream:
+        # Streaming mode - show output as it generates
+        full_response = ""
+        final_code = ""
+        
+        console.print("[dim]Streaming response...[/dim]\n")
+        
+        for chunk, is_complete in generator.generate_stream(
+            description=description,
+            mode=mode,
+            language=language,
+            name=name or "",
+            parameters=params or "",
+            source_code=source_code,
+        ):
+            if is_complete:
+                final_code = chunk
+            else:
+                console.print(chunk, end="", highlight=False)
+                full_response += chunk
+        
+        console.print("\n")
+        
+        # Create result for formatting
+        from ai_code_assistant.generator import GenerationResult
+        result = GenerationResult(
+            code=final_code,
+            language=language,
+            mode=mode,
+            description=description,
+            raw_response=full_response,
+        )
+        
+        # Show extracted code in a panel
+        console.print(Panel(
+            final_code,
+            title=f"[bold green]Generated {mode.title()}[/bold green]",
+            border_style="green",
+        ))
+    else:
+        # Non-streaming mode (original behavior)
+        with console.status("[bold green]Generating code..."):
+            if mode == "function":
+                result = generator.generate_function(
+                    description=description, name=name or "generated_function",
+                    language=language, parameters=params or "",
+                )
+            elif mode == "class":
+                result = generator.generate_class(
+                    description=description, name=name or "GeneratedClass", language=language,
+                )
+            elif mode == "script":
+                result = generator.generate_script(
+                    description=description, requirements=[description], language=language,
+                )
+            elif mode == "test":
+                result = generator.generate_tests(source_code=source_code, language=language)
+            else:
+                result = generator.generate(description=description, language=language)
 
-    formatted = formatter.format_generation(result)
+        formatted = formatter.format_generation(result)
 
     if output and result.success:
         output.parent.mkdir(parents=True, exist_ok=True)

ai_code_assistant/generator/code_gen.py

@@ -3,7 +3,7 @@
 import re
 from dataclasses import dataclass, field
 from pathlib import Path
-from typing import List, Literal, Optional
+from typing import Iterator, List, Literal, Optional, Tuple
 
 from ai_code_assistant.config import Config
 from ai_code_assistant.llm import LLMManager
@@ -214,6 +214,160 @@ class CodeGenerator:
                 error=str(e),
             )
 
+
+    def generate_stream(
+        self,
+        description: str,
+        mode: str = "generic",
+        language: str = "python",
+        name: str = "",
+        parameters: str = "",
+        source_code: str = "",
+    ) -> Iterator[Tuple[str, bool]]:
+        """
+        Stream code generation with real-time output.
+        
+        Yields tuples of (chunk, is_complete).
+        The final yield will have is_complete=True and contain the extracted code.
+        """
+        # Build the prompt based on mode
+        if mode == "function":
+            prompt_template = GENERATION_PROMPTS["function"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                name=name or "generated_function",
+                parameters=parameters or "None specified",
+                return_type="appropriate type",
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+        elif mode == "class":
+            prompt_template = GENERATION_PROMPTS["class"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                name=name or "GeneratedClass",
+                attributes="None specified",
+                methods="None specified",
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+        elif mode == "script":
+            prompt_template = GENERATION_PROMPTS["script"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                requirements=f"- {description}",
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+        elif mode == "test":
+            prompt_template = GENERATION_PROMPTS["test"]
+            formatted = prompt_template.format(
+                language=language,
+                source_code=source_code,
+                test_framework="pytest",
+            )
+        else:  # generic
+            prompt_template = GENERATION_PROMPTS["generic"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+
+        # Stream the response
+        full_response = ""
+        try:
+            for chunk in self.llm.stream(formatted):
+                full_response += chunk
+                yield (chunk, False)
+            
+            # Extract code and yield final result
+            code = self._extract_code(full_response, language)
+            yield (code, True)
+        except Exception as e:
+            yield (f"Error: {str(e)}", True)
+
+
+    def generate_stream(
+        self,
+        description: str,
+        mode: str = "generic",
+        language: str = "python",
+        name: str = "",
+        parameters: str = "",
+        source_code: str = "",
+    ) -> Iterator[Tuple[str, bool]]:
+        """
+        Stream code generation with real-time output.
+        
+        Yields tuples of (chunk, is_complete).
+        The final yield will have is_complete=True and contain the extracted code.
+        """
+        # Build the prompt based on mode
+        if mode == "function":
+            prompt_template = GENERATION_PROMPTS["function"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                name=name or "generated_function",
+                parameters=parameters or "None specified",
+                return_type="appropriate type",
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+        elif mode == "class":
+            prompt_template = GENERATION_PROMPTS["class"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                name=name or "GeneratedClass",
+                attributes="None specified",
+                methods="None specified",
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+        elif mode == "script":
+            prompt_template = GENERATION_PROMPTS["script"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                requirements=f"- {description}",
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+        elif mode == "test":
+            prompt_template = GENERATION_PROMPTS["test"]
+            formatted = prompt_template.format(
+                language=language,
+                source_code=source_code,
+                test_framework="pytest",
+            )
+        else:  # generic
+            prompt_template = GENERATION_PROMPTS["generic"]
+            formatted = prompt_template.format(
+                language=language,
+                description=description,
+                include_type_hints=self.config.generation.include_type_hints,
+                include_docstrings=self.config.generation.include_docstrings,
+            )
+
+        # Stream the response
+        full_response = ""
+        try:
+            for chunk in self.llm.stream(formatted):
+                full_response += chunk
+                yield (chunk, False)
+            
+            # Extract code and yield final result
+            code = self._extract_code(full_response, language)
+            yield (code, True)
+        except Exception as e:
+            yield (f"Error: {str(e)}", True)
+
     def _extract_code(self, response: str, language: str) -> str:
         """Extract code block from LLM response."""
         # Try to find language-specific code block

{cognify_code-0.2.2.dist-info → cognify_code-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognify-code
-Version: 0.2.2
+Version: 0.2.3
 Summary: Your local AI-powered code assistant. Review, generate, search, and refactor code with an intelligent AI agent—all running locally with complete privacy.
 Author-email: Ashok Kumar <akkssy@users.noreply.github.com>
 Maintainer-email: Ashok Kumar <akkssy@users.noreply.github.com>

{cognify_code-0.2.2.dist-info → cognify_code-0.2.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ai_code_assistant/__init__.py,sha256=XnpG4h-2gW3cXseFvqQT_-XyOmVJtikVMrHUnmy8XKI,409
-ai_code_assistant/cli.py,sha256=47TI6XMkcgojYLSLADnk3BphxUffSgmV0tCTHIu4W7o,61350
+ai_code_assistant/cli.py,sha256=lvgNRB0LwGmZJbUUJHRdZMuoV83USMvNUB8-wLZjBkc,62987
 ai_code_assistant/config.py,sha256=6sAufexwzfCu2JNWvt9KevS9k_gMcjj1TAnwuaO1ZFw,4727
 ai_code_assistant/llm.py,sha256=DfcWJf6zEAUsPSEZLdEmb9o6BQNf1Ja88nswjpy6cOw,4209
 ai_code_assistant/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -18,7 +18,7 @@ ai_code_assistant/editor/diff_handler.py,sha256=LeI-00GuH7ASIetsUzT3Y_pDq4K1wmyc
 ai_code_assistant/editor/file_editor.py,sha256=csD8MW0jrfXAek5blWNuot_QWlhkgTTmtQtf8rbIdhY,11143
 ai_code_assistant/editor/prompts.py,sha256=wryxwb4dNaeSbl5mHkDuw2uTAZxkdrdyZ89Gfogafow,4356
 ai_code_assistant/generator/__init__.py,sha256=CfCO58CBye-BlZHjOfLLShovp2TVXg_GHKJuXe6ihu0,273
-ai_code_assistant/generator/code_gen.py,sha256=vJ4xeiG4_1hOZ8TW7YZ-gTAVKoiHwgUzzdcWMvhgeRo,8609
+ai_code_assistant/generator/code_gen.py,sha256=Sp_j1IdR0vbU5xRZHhvTBCbxfNxtvNMU1tEg0NLV00k,14790
 ai_code_assistant/generator/prompts.py,sha256=uoEDpcRzpTd-4TLHNW_EbSHJiADMlu9SoGWZvvo1Adk,3384
 ai_code_assistant/git/__init__.py,sha256=YgqmzneAnZyRrbazMqGoFSPIk5Yf5OTm2LXPbkQmecU,232
 ai_code_assistant/git/commit_generator.py,sha256=CzDH5ZPqEaXyPznBg8FgTz8wbV4adALUQD__kl8au6o,4135
@@ -47,9 +47,9 @@ ai_code_assistant/reviewer/prompts.py,sha256=9RrHEBttS5ngxY2BNsUvqGC6-cTxco-kDPb
 ai_code_assistant/utils/__init__.py,sha256=3HO-1Bj4VvUtM7W1C3MKR4DzQ9Xc875QKSHHkHwuqVs,368
 ai_code_assistant/utils/file_handler.py,sha256=jPxvtI5dJxkpPjELgRJ11WXamtyKKmZANQ1fcfMVtiU,5239
 ai_code_assistant/utils/formatters.py,sha256=5El9ew9HS6JLBucBUxxcw4fO5nLpOucgNJrJj2NC3zw,8945
-cognify_code-0.2.2.dist-info/licenses/LICENSE,sha256=5yu_kWq2bK-XKhWo79Eykdg4Qf3O8V2Ys7cpOO7GyyE,1063
-cognify_code-0.2.2.dist-info/METADATA,sha256=HRjZsc18DNcoS-NOJw9mCXB3qvrwuP1GhkgeLHmucOg,11862
-cognify_code-0.2.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-cognify_code-0.2.2.dist-info/entry_points.txt,sha256=MrBnnWPHZVozqqKyTlnJO63YN2kE5yPWKlr2nnRFRks,94
-cognify_code-0.2.2.dist-info/top_level.txt,sha256=dD_r1x-oX0s1uspYY72kig4jfIsjh3oDKwOBCMYXqpo,18
-cognify_code-0.2.2.dist-info/RECORD,,
+cognify_code-0.2.3.dist-info/licenses/LICENSE,sha256=5yu_kWq2bK-XKhWo79Eykdg4Qf3O8V2Ys7cpOO7GyyE,1063
+cognify_code-0.2.3.dist-info/METADATA,sha256=oGRohUiX2gCBNbV7FqAWyRlK_nD7aGFav2AjCuESJo0,11862
+cognify_code-0.2.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+cognify_code-0.2.3.dist-info/entry_points.txt,sha256=MrBnnWPHZVozqqKyTlnJO63YN2kE5yPWKlr2nnRFRks,94
+cognify_code-0.2.3.dist-info/top_level.txt,sha256=dD_r1x-oX0s1uspYY72kig4jfIsjh3oDKwOBCMYXqpo,18
+cognify_code-0.2.3.dist-info/RECORD,,