pdd-cli 0.0.45__py3-none-any.whl → 0.0.118__py3-none-any.whl

pdd/unfinished_prompt.py

@@ -1,4 +1,5 @@
-from typing import Tuple
+from typing import Tuple, Optional
+import ast
 from pydantic import BaseModel, Field
 from rich import print as rprint
 from .load_prompt_template import load_prompt_template
@@ -14,6 +15,7 @@ def unfinished_prompt(
     strength: float = DEFAULT_STRENGTH,
     temperature: float = 0,
     time: float = DEFAULT_TIME,
+    language: Optional[str] = None,
     verbose: bool = False
 ) -> Tuple[str, bool, float, str]:
     """
@@ -48,6 +50,40 @@ def unfinished_prompt(
         if not 0 <= temperature <= 1:
             raise ValueError("Temperature must be between 0 and 1")
 
+        # Step 0: Fast syntactic completeness check for Python tails
+        # Apply when language explicitly 'python' or when the text likely looks like Python.
+        def _looks_like_python(text: str) -> bool:
+            lowered = text.strip().lower()
+            py_signals = (
+                "def ", "class ", "import ", "from ",
+            )
+            if any(sig in lowered for sig in py_signals):
+                return True
+            # Heuristic: has 'return ' without JS/TS markers
+            if "return " in lowered and not any(tok in lowered for tok in ("function", "=>", ";", "{", "}")):
+                return True
+            # Heuristic: colon-introduced blocks and indentation
+            if ":\n" in text or "\n    " in text:
+                return True
+            return False
+
+        should_try_python_parse = (language or "").lower() == "python" or _looks_like_python(prompt_text)
+        if should_try_python_parse:
+            try:
+                ast.parse(prompt_text)
+                reasoning = "Syntactic Python check passed (ast.parse succeeded); treating as finished."
+                if verbose:
+                    rprint("[green]" + reasoning + "[/green]")
+                return (
+                    reasoning,
+                    True,
+                    0.0,
+                    "syntactic_check"
+                )
+            except SyntaxError:
+                # Fall through to LLM-based judgment
+                pass
+
         # Step 1: Load the prompt template
         if verbose:
            rprint("[blue]Loading prompt template...[/blue]")
@@ -58,6 +94,9 @@ def unfinished_prompt(
 
         # Step 2: Prepare input and invoke LLM
         input_json = {"PROMPT_TEXT": prompt_text}
+        # Optionally pass a language hint to the prompt
+        if language:
+            input_json["LANGUAGE"] = language
         
         if verbose:
             rprint("[blue]Invoking LLM model...[/blue]")
@@ -75,14 +114,22 @@ def unfinished_prompt(
             temperature=temperature,
             time=time,
             verbose=verbose,
-            output_pydantic=PromptAnalysis
+            output_pydantic=PromptAnalysis,
+            language=language,
         )
 
         # Step 3: Extract and return results
-        result: PromptAnalysis = response['result']
+        result = response['result']
         total_cost = response['cost']
         model_name = response['model_name']
 
+        # Defensive type checking: ensure we got a PromptAnalysis, not a raw string
+        if not isinstance(result, PromptAnalysis):
+            raise TypeError(
+                f"Expected PromptAnalysis from llm_invoke, got {type(result).__name__}. "
+                f"This typically indicates JSON parsing failed. Value: {repr(result)[:200]}"
+            )
+
         if verbose:
            rprint("[green]Analysis complete![/green]")
            rprint(f"Reasoning: {result.reasoning}")
@@ -116,4 +163,4 @@ if __name__ == "__main__":
         rprint(f"Cost: ${cost:.6f}")
         rprint(f"Model: {model}")
     except Exception as e:
-        rprint("[red]Error in example:[/red]", str(e))
+        rprint("[red]Error in example:[/red]", str(e))