pdd-cli 0.0.24__py3-none-any.whl → 0.0.25__py3-none-any.whl

pdd/llm_invoke.py

@@ -25,6 +25,7 @@ import json
 
 from pydantic import BaseModel, Field
 from rich import print as rprint
+from rich.errors import MarkupError
 
 # Langchain core and community imports
 from langchain_core.prompts import PromptTemplate, ChatPromptTemplate
@@ -347,12 +348,22 @@ def llm_invoke(prompt, input_json, strength, temperature, verbose=False, output_
                 rprint(f"Strength used: {strength}")
                 rprint(f"Temperature used: {temperature}")
                 try:
-                    rprint(f"Input JSON: {str(input_json)}")  # Use str() instead of json.dumps()
+                    # Try printing with rich formatting first
+                    rprint(f"Input JSON: {str(input_json)}")
+                except MarkupError:
+                    # Fallback to standard print if rich markup fails
+                    print(f"Input JSON: {str(input_json)}")
                 except Exception:
-                    rprint(f"Input JSON: {input_json}")
+                    print(f"Input JSON: {input_json}")
                 if output_pydantic:
                     rprint(f"Output Pydantic format: {output_pydantic}")
-                rprint(f"Result: {result_output}")
+                try:
+                    # Try printing with rich formatting first
+                    rprint(f"Result: {result_output}")
+                except MarkupError as me:
+                    # Fallback to standard print if rich markup fails
+                    print(f"[bold yellow]Warning:[/bold yellow] Failed to render result with rich markup: {me}")
+                    print(f"Raw Result: {str(result_output)}") # Use standard print
 
             return {'result': result_output, 'cost': cost, 'model_name': model.model}
 

pdd/preprocess.py

@@ -133,9 +133,9 @@ def process_web_tags(text: str) -> str:
                 console.print("[bold yellow]Warning:[/bold yellow] FIRECRAWL_API_KEY not found in environment")
                 return f"[Error: FIRECRAWL_API_KEY not set. Cannot scrape {url}]"
             app = FirecrawlApp(api_key=api_key)
-            response = app.scrape_url(url=url, params={'formats': ['markdown']})
-            if 'markdown' in response:
-                return response['markdown']
+            response = app.scrape_url(url, formats=['markdown'])
+            if hasattr(response, 'markdown'):
+                return response.markdown
             else:
                 console.print(f"[bold yellow]Warning:[/bold yellow] No markdown content returned for {url}")
                 return f"[No content available for {url}]"