923fa671fe
ContentGuardianAgent consolidation:
- Merge 3 duplicate classes into single source in specialized/content_guardian.py
- Watchdog audit_committee() with heuristic scoring, coverage gaps, overlaps, alerts
- Remove misleading rejection_rate() helper; use acceptance_rate directly
- Integrate audit + alerts + trend signals into today_workflow_service.py
Team Activity page:
- QualityAuditPanel: health ring, per-agent critiques, coverage gaps, overlaps
- TrendSignalsPanel: opportunity cards with urgency/impact/coverage bars
- AlertBanner: persistent dismiss via POST /alerts/{id}/mark-read
- AgentHelpModal: dialog showing all 8 agents with descriptions, tools, schedule
- QualityAuditPanel action buttons: Fill gap -> /content-planning, Resolve overlap, View CTA on alerts/issues
- TrendSignalsPanel action buttons: Create content from this trend -> /blog-writer with trend context state
Onboarding system:
- Step 4 validation: no auto-pass via basic_ready; requires persona data or explicit progression
- Step 5 validation: logs warning on auto-pass without integration data
- OnboardingCompletionService: single DB session, transactional task creation, upsert pattern
- Business-without-website: nullable website_url on SIFIndexingTask and MarketTrendsTask
- DeepCompetitorAnalysisExecutor: 5-min timeout, 10-competitor cap, asyncio.wait_for
- Persona generation: async with 30s timeout, falls back to scheduler
- OnboardingProgressService.reset_onboarding(): resets session + pauses all DB tasks
- OnboardingControlService.reset_onboarding(): also cancels APScheduler jobs
- FinalStep TaskSchedulingPanel: shows scheduled/failed tasks after completion, 8s auto-redirect
- onboarding_completed agent activity event logged to feed
Documentation:
- docs-site/features/onboarding/: overview, steps, scheduler-tasks, technical-reference (4 pages)
- docs-site/mkdocs.yml: added Onboarding System nav section
- docs-site/features/sif-agents/: overview, agent-directory, committee-system, content-guardian (4 pages)
- docs-site/features/team-activity/: overview, quality-audit, trend-signals, alert-system (4 pages)
- docs-site/features/todays-workflow/: updated overview, technical-architecture, workflow-guide, api-reference
646 lines
26 KiB
Python
646 lines
26 KiB
Python
"""
|
|
WaveSpeed LLM Provider Module for ALwrity
|
|
|
|
This module provides functions for interacting with WaveSpeed's LLM API
|
|
using the OpenAI-compatible interface for text generation.
|
|
|
|
Key Features:
|
|
- Text response generation with retry logic
|
|
- Comprehensive error handling and logging
|
|
- Automatic API key management
|
|
- Support for gpt-oss and other WaveSpeed models
|
|
- Integration with subscription/preflight checks
|
|
|
|
Best Practices:
|
|
1. Use appropriate temperature for your use case (0.7 for creative, 0.1-0.3 for factual)
|
|
2. Set max_tokens based on expected response length
|
|
3. Use system_prompt to guide model behavior
|
|
4. Handle errors gracefully in calling functions
|
|
|
|
Usage Examples:
|
|
# Text response
|
|
result = wavespeed_text_response(prompt, temperature=0.7, max_tokens=2048)
|
|
|
|
# Structured JSON response
|
|
schema = {"type": "object", "properties": {"title": {"type": "string"}}}
|
|
result = wavespeed_structured_json_response(prompt, schema, temperature=0.2, max_tokens=8192)
|
|
|
|
Dependencies:
|
|
- openai (for WaveSpeed OpenAI-compatible API)
|
|
- tenacity (for retry logic)
|
|
- logging (for debugging)
|
|
- json (for fallback parsing)
|
|
|
|
Author: ALwrity Team
|
|
Version: 1.0
|
|
Last Updated: March 2026
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import time as _time
|
|
from pathlib import Path
|
|
import json
|
|
import re
|
|
from typing import Optional, Dict, Any, List
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
# Fix the environment loading path - load from backend directory
|
|
_mod_start = _time.time()
|
|
current_dir = Path(__file__).parent.parent # services directory
|
|
backend_dir = current_dir.parent # backend directory
|
|
env_path = backend_dir / '.env'
|
|
|
|
if env_path.exists():
|
|
load_dotenv(env_path)
|
|
_dotenv_ms = (_time.time() - _mod_start) * 1000
|
|
print(f"Loaded .env from: {env_path} (took {_dotenv_ms:.0f}ms)")
|
|
else:
|
|
load_dotenv()
|
|
print(f"No .env found at {env_path}, using current directory")
|
|
|
|
from loguru import logger
|
|
from utils.logger_utils import get_service_logger
|
|
|
|
# Use service-specific logger to avoid conflicts
|
|
logger = get_service_logger("wavespeed_provider")
|
|
|
|
_import_start = _time.time()
|
|
from tenacity import (
|
|
retry,
|
|
retry_if_exception,
|
|
stop_after_attempt,
|
|
wait_random_exponential,
|
|
)
|
|
|
|
try:
|
|
from openai import OpenAI
|
|
from openai import NotFoundError
|
|
OPENAI_AVAILABLE = True
|
|
except ImportError:
|
|
OPENAI_AVAILABLE = False
|
|
NotFoundError = Exception
|
|
logger.warn("OpenAI library not available. Install with: pip install openai")
|
|
|
|
logger.warning(f"[wavespeed_provider] Module import completed in {(_time.time()-_import_start)*1000:.0f}ms (openai_available={OPENAI_AVAILABLE})")
|
|
|
|
# Default WaveSpeed models for fallback
|
|
WAVESPEED_FALLBACK_MODELS = [
|
|
"openai/gpt-oss-120b",
|
|
"meta-llama/Llama-3.1-8B-Instruct",
|
|
"mistralai/Mistral-7B-Instruct-v0.3",
|
|
"google/gemma-7b-it",
|
|
]
|
|
|
|
def _candidate_model_variants(model: str):
|
|
"""Yield model ids to try for a single logical model preference."""
|
|
if not model:
|
|
return
|
|
|
|
# Try configured model first
|
|
yield model
|
|
|
|
# Fallback to base repo id when provider suffix is not recognized by the router
|
|
if ":" in model:
|
|
base_model = model.split(":", 1)[0]
|
|
if base_model:
|
|
yield base_model
|
|
|
|
def _fallback_model_sequence(model: str, fallback_models: Optional[List[str]] = None):
|
|
# IMPORTANT: Do not apply implicit global fallback chains.
|
|
# Callers must explicitly provide fallback_models when they want multi-model retries.
|
|
if fallback_models:
|
|
sequence = [model] + fallback_models
|
|
else:
|
|
sequence = [model]
|
|
seen = set()
|
|
for preferred_model in sequence:
|
|
for candidate in _candidate_model_variants(preferred_model):
|
|
if candidate and candidate not in seen:
|
|
seen.add(candidate)
|
|
yield candidate
|
|
|
|
def _is_non_retryable_wavespeed_error(exc: Exception) -> bool:
|
|
"""Skip retries for deterministic WaveSpeed failures (e.g., unknown model ids, billing)."""
|
|
msg = str(exc).lower()
|
|
status = getattr(exc, "status_code", None)
|
|
|
|
# Non-retryable errors
|
|
if isinstance(exc, NotFoundError) or "not found" in msg or "404" in msg:
|
|
return True
|
|
if status == 402 or "402" in msg or "depleted" in msg or "credits" in msg:
|
|
return True
|
|
if status == 401 or "unauthorized" in msg or "401" in msg:
|
|
return True
|
|
if status == 403 or "forbidden" in msg or "403" in msg:
|
|
return True
|
|
|
|
return False
|
|
|
|
def _should_retry_wavespeed_error(exc: Exception) -> bool:
|
|
return not _is_non_retryable_wavespeed_error(exc)
|
|
|
|
def _classify_wavespeed_error(exc: Exception) -> str:
|
|
"""Classify WaveSpeed failures for actionable logs."""
|
|
msg = str(exc).lower()
|
|
if any(token in msg for token in ["insufficient", "balance", "quota", "billing", "payment", "402"]):
|
|
return "billing_or_quota"
|
|
if "unauthorized" in msg or "forbidden" in msg or "401" in msg or "403" in msg:
|
|
return "auth_or_permission"
|
|
if "not found" in msg or "404" in msg:
|
|
return "model_not_found"
|
|
return "unknown"
|
|
|
|
def _wavespeed_error_details(exc: Exception) -> str:
|
|
"""Return compact, actionable exception details for logs."""
|
|
status = getattr(exc, "status_code", None)
|
|
err_type = type(exc).__name__
|
|
message = str(exc)
|
|
raw_body = getattr(exc, "body", None)
|
|
details = f"type={err_type}"
|
|
if status is not None:
|
|
details += f", status={status}"
|
|
if message:
|
|
details += f", message={message}"
|
|
if raw_body:
|
|
details += f", body={raw_body}"
|
|
details += f", repr={repr(exc)}"
|
|
return details
|
|
|
|
def get_wavespeed_api_key() -> str:
|
|
"""Get WaveSpeed API key with proper error handling."""
|
|
api_key = os.getenv('WAVESPEED_API_KEY')
|
|
if not api_key:
|
|
error_msg = "WAVESPEED_API_KEY environment variable is not set. Please set it in your .env file."
|
|
logger.error(error_msg)
|
|
raise ValueError(error_msg)
|
|
|
|
# Validate API key format (basic check)
|
|
if not api_key or len(api_key) < 10:
|
|
error_msg = "WAVESPEED_API_KEY appears to be invalid."
|
|
logger.error(error_msg)
|
|
raise ValueError(error_msg)
|
|
|
|
return api_key
|
|
|
|
|
|
def _retry_with_increased_tokens(
|
|
client: "OpenAI",
|
|
messages: List[Dict[str, str]],
|
|
model: str,
|
|
fallback_models: Optional[List[str]],
|
|
temperature: float,
|
|
max_tokens: int,
|
|
) -> Optional[str]:
|
|
"""Retry the API call with increased max_tokens when JSON parsing fails due to truncation."""
|
|
max_tokens = min(max_tokens, 16384)
|
|
last_error = None
|
|
for candidate_model in _fallback_model_sequence(model, fallback_models):
|
|
try:
|
|
response = client.chat.completions.create(
|
|
model=candidate_model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens,
|
|
)
|
|
text = response.choices[0].message.content
|
|
text = text.strip() if text else ""
|
|
if text.startswith("```json"):
|
|
text = text[7:]
|
|
if text.startswith("```"):
|
|
text = text[3:]
|
|
if text.endswith("```"):
|
|
text = text[:-3]
|
|
return text.strip()
|
|
except NotFoundError as nf_err:
|
|
last_error = nf_err
|
|
continue
|
|
if last_error:
|
|
logger.error(f"All fallback models failed on retry with increased tokens: {last_error}")
|
|
return None
|
|
|
|
|
|
@retry(
|
|
retry=retry_if_exception(_should_retry_wavespeed_error),
|
|
wait=wait_random_exponential(min=1, max=60),
|
|
stop=stop_after_attempt(6),
|
|
)
|
|
def wavespeed_text_response(
|
|
prompt: str,
|
|
model: str = "openai/gpt-oss-120b",
|
|
fallback_models: Optional[List[str]] = None,
|
|
temperature: float = 0.7,
|
|
max_tokens: int = 2048,
|
|
top_p: float = 0.9,
|
|
system_prompt: Optional[str] = None
|
|
) -> str:
|
|
"""
|
|
Generate text response using WaveSpeed LLM API.
|
|
|
|
This function uses the WaveSpeed OpenAI-compatible API for text generation
|
|
with built-in retry logic and error handling.
|
|
|
|
Args:
|
|
prompt (str): The input prompt for the AI model
|
|
model (str): WaveSpeed model identifier (default: "openai/gpt-oss-120b")
|
|
temperature (float): Controls randomness (0.0-1.0)
|
|
max_tokens (int): Maximum tokens in response
|
|
top_p (float): Nucleus sampling parameter (0.0-1.0)
|
|
system_prompt (str, optional): System instruction for the model
|
|
|
|
Returns:
|
|
str: Generated text response
|
|
|
|
Raises:
|
|
Exception: If API key is missing or API call fails
|
|
|
|
Best Practices:
|
|
- Use appropriate temperature for your use case (0.7 for creative, 0.1-0.3 for factual)
|
|
- Set max_tokens based on expected response length
|
|
- Use system_prompt to guide model behavior
|
|
- Handle errors gracefully in calling functions
|
|
|
|
Example:
|
|
result = wavespeed_text_response(
|
|
prompt="Write a blog post about AI",
|
|
model="openai/gpt-oss-120b",
|
|
temperature=0.7,
|
|
max_tokens=2048,
|
|
system_prompt="You are a professional content writer."
|
|
)
|
|
"""
|
|
try:
|
|
if not OPENAI_AVAILABLE:
|
|
raise ImportError("OpenAI library not available. Install with: pip install openai")
|
|
|
|
# Get API key with proper error handling
|
|
api_key = get_wavespeed_api_key()
|
|
logger.info(f"🔑 WaveSpeed API key loaded: {bool(api_key)} (length: {len(api_key) if api_key else 0})")
|
|
|
|
if not api_key:
|
|
raise Exception("WAVESPEED_API_KEY not found in environment variables")
|
|
|
|
_t0 = _time.time()
|
|
# Initialize WaveSpeed client
|
|
client = OpenAI(
|
|
base_url="https://llm.wavespeed.ai/v1",
|
|
api_key=api_key,
|
|
)
|
|
logger.warning(f"[wavespeed_text_response] OpenAI client init took {(_time.time()-_t0)*1000:.0f}ms")
|
|
|
|
# Prepare input for the API
|
|
messages = []
|
|
|
|
# Add system prompt if provided
|
|
if system_prompt:
|
|
messages.append({
|
|
"role": "system",
|
|
"content": system_prompt
|
|
})
|
|
|
|
# Add user prompt
|
|
messages.append({
|
|
"role": "user",
|
|
"content": prompt
|
|
})
|
|
|
|
# Add debugging for API call
|
|
logger.info(
|
|
"WaveSpeed text call | model={} | prompt_len={} | temp={} | top_p={} | max_tokens={}",
|
|
model,
|
|
len(prompt) if isinstance(prompt, str) else '<non-str>',
|
|
temperature,
|
|
top_p,
|
|
max_tokens,
|
|
)
|
|
|
|
logger.info("🚀 Making WaveSpeed API call (chat completion)...")
|
|
|
|
_api_t0 = _time.time()
|
|
# Call exactly the requested model; no retries, no fallbacks, no variants
|
|
response = client.chat.completions.create(
|
|
model=model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
top_p=top_p,
|
|
max_tokens=max_tokens
|
|
)
|
|
logger.warning(f"[wavespeed_text_response] API call took {(_time.time()-_api_t0)*1000:.0f}ms")
|
|
|
|
# Extract text from response
|
|
generated_text = response.choices[0].message.content
|
|
|
|
# Clean up the response
|
|
if generated_text:
|
|
# Remove any markdown formatting if present
|
|
generated_text = re.sub(r'```[a-zA-Z]*\n?', '', generated_text)
|
|
generated_text = re.sub(r'```\n?', '', generated_text)
|
|
generated_text = generated_text.strip()
|
|
|
|
logger.info(f"✅ WaveSpeed text response generated successfully (length: {len(generated_text)})")
|
|
return generated_text
|
|
|
|
except Exception as e:
|
|
error_class = _classify_wavespeed_error(e)
|
|
error_details = _wavespeed_error_details(e)
|
|
logger.error(f"❌ WaveSpeed text generation failed: {error_details}")
|
|
|
|
# Extra diagnostics: try to capture raw response if available
|
|
if hasattr(e, 'response') and e.response is not None:
|
|
logger.error(f"🔍 WaveSpeed Error Diagnostics:")
|
|
logger.error(f" - Status: {e.response.status_code}")
|
|
logger.error(f" - Headers: {dict(e.response.headers)}")
|
|
try:
|
|
body_json = e.response.json()
|
|
logger.error(f" - Body JSON: {json.dumps(body_json, indent=2)}")
|
|
except Exception:
|
|
logger.error(f" - Body Raw: {e.response.text[:1000]}")
|
|
else:
|
|
logger.error(f"🔍 No HTTP response attached to exception object.")
|
|
|
|
raise Exception(f"WaveSpeed text generation failed: {str(e)}")
|
|
|
|
@retry(
|
|
retry=retry_if_exception(_should_retry_wavespeed_error),
|
|
wait=wait_random_exponential(min=1, max=60),
|
|
stop=stop_after_attempt(6),
|
|
)
|
|
def wavespeed_structured_json_response(
|
|
prompt: str,
|
|
schema: Dict[str, Any],
|
|
model: str = "openai/gpt-oss-120b",
|
|
fallback_models: Optional[List[str]] = None,
|
|
temperature: float = 0.7,
|
|
max_tokens: int = 8192,
|
|
system_prompt: Optional[str] = None
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Generate structured JSON response using WaveSpeed LLM API.
|
|
|
|
This function uses the WaveSpeed OpenAI-compatible API with structured output support
|
|
to generate JSON responses that match a provided schema.
|
|
|
|
Args:
|
|
prompt (str): The input prompt for the AI model
|
|
schema (dict): JSON schema defining the expected output structure
|
|
model (str): WaveSpeed model identifier (default: "openai/gpt-oss-120b")
|
|
temperature (float): Controls randomness (0.0-1.0). Use 0.1-0.3 for structured output
|
|
max_tokens (int): Maximum tokens in response. Use 8192 for complex outputs
|
|
system_prompt (str, optional): System instruction for the model
|
|
|
|
Returns:
|
|
dict: Parsed JSON response matching the provided schema
|
|
|
|
Raises:
|
|
Exception: If API key is missing or API call fails
|
|
|
|
Best Practices:
|
|
- Keep schemas simple and flat to avoid truncation
|
|
- Use low temperature (0.1-0.3) for consistent structured output
|
|
- Set max_tokens to 8192 for complex multi-field responses
|
|
- Avoid deeply nested schemas with many required fields
|
|
- Test with smaller outputs first, then scale up
|
|
|
|
Example:
|
|
schema = {
|
|
"type": "object",
|
|
"properties": {
|
|
"tasks": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "object",
|
|
"properties": {
|
|
"title": {"type": "string"},
|
|
"description": {"type": "string"}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
result = wavespeed_structured_json_response(prompt, schema, temperature=0.2, max_tokens=8192)
|
|
"""
|
|
try:
|
|
if not OPENAI_AVAILABLE:
|
|
raise ImportError("OpenAI library not available. Install with: pip install openai")
|
|
|
|
# Get API key with proper error handling
|
|
api_key = get_wavespeed_api_key()
|
|
logger.info(f"🔑 WaveSpeed API key loaded: {bool(api_key)} (length: {len(api_key) if api_key else 0})")
|
|
|
|
if not api_key:
|
|
raise Exception("WAVESPEED_API_KEY not found in environment variables")
|
|
|
|
_fn_start = _time.time()
|
|
# Initialize OpenAI client with WaveSpeed base URL
|
|
client = OpenAI(
|
|
base_url="https://llm.wavespeed.ai/v1",
|
|
api_key=api_key,
|
|
)
|
|
_client_init_ms = (_time.time() - _fn_start) * 1000
|
|
logger.warning(f"[wavespeed_structured_json_response] OpenAI client init took {_client_init_ms:.0f}ms")
|
|
|
|
# Prepare input for the API
|
|
messages = []
|
|
|
|
# Add system prompt if provided
|
|
if system_prompt:
|
|
messages.append({
|
|
"role": "system",
|
|
"content": system_prompt
|
|
})
|
|
|
|
# Add user prompt with JSON instruction
|
|
json_instruction = "Please respond with valid JSON that matches the provided schema."
|
|
messages.append({
|
|
"role": "user",
|
|
"content": f"{prompt}\n\n{json_instruction}"
|
|
})
|
|
|
|
# Add debugging for API call
|
|
logger.info(
|
|
"WaveSpeed structured call | model={} | prompt_len={} | schema_kind={} | temp={} | max_tokens={}",
|
|
model,
|
|
len(prompt) if isinstance(prompt, str) else '<non-str>',
|
|
type(schema).__name__,
|
|
temperature,
|
|
max_tokens,
|
|
)
|
|
|
|
logger.info("🚀 Making WaveSpeed structured API call...")
|
|
|
|
# Add JSON schema to prompt for guidance
|
|
json_schema_str = json.dumps(schema, indent=2)
|
|
messages[-1]["content"] += f"\n\nJSON Schema:\n{json_schema_str}"
|
|
|
|
_api_start = _time.time()
|
|
try:
|
|
response = None
|
|
last_error = None
|
|
for candidate_model in _fallback_model_sequence(model, fallback_models):
|
|
try:
|
|
logger.info(f"[wavespeed_structured_json_response] Calling model={candidate_model}...")
|
|
response = client.chat.completions.create(
|
|
model=candidate_model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens,
|
|
response_format={"type": "json_object"} # Try to enforce JSON mode if supported
|
|
)
|
|
_api_ms = (_time.time() - _api_start) * 1000
|
|
if candidate_model != model:
|
|
logger.warning("WaveSpeed structured generation switched to fallback model: {}", candidate_model)
|
|
logger.warning(f"[wavespeed_structured_json_response] First API call completed in {_api_ms:.0f}ms (model={candidate_model})")
|
|
break
|
|
except NotFoundError as nf_err:
|
|
last_error = nf_err
|
|
logger.warning("WaveSpeed structured model not found: {}. Trying fallback model.", candidate_model)
|
|
continue
|
|
|
|
if response is None:
|
|
raise last_error or Exception("WaveSpeed structured generation failed: all fallback models failed")
|
|
|
|
response_text = response.choices[0].message.content
|
|
response_text = response_text.strip() if response_text else ""
|
|
|
|
# If response_format returned empty content, retry without it
|
|
if not response_text:
|
|
logger.warning("WaveSpeed structured call returned empty content with response_format, retrying without it...")
|
|
response = None
|
|
last_error = None
|
|
for candidate_model in _fallback_model_sequence(model, fallback_models):
|
|
try:
|
|
response = client.chat.completions.create(
|
|
model=candidate_model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens
|
|
)
|
|
break
|
|
except NotFoundError as nf_err:
|
|
last_error = nf_err
|
|
continue
|
|
if response is not None:
|
|
response_text = response.choices[0].message.content
|
|
response_text = response_text.strip() if response_text else ""
|
|
|
|
# Clean up response text if needed
|
|
if response_text.startswith("```json"):
|
|
response_text = response_text[7:]
|
|
if response_text.startswith("```"):
|
|
response_text = response_text[3:]
|
|
if response_text.endswith("```"):
|
|
response_text = response_text[:-3]
|
|
response_text = response_text.strip()
|
|
|
|
try:
|
|
parsed_json = json.loads(response_text) if response_text else None
|
|
if parsed_json is not None:
|
|
logger.info("✅ WaveSpeed structured JSON response parsed successfully")
|
|
return parsed_json
|
|
except json.JSONDecodeError as json_err:
|
|
logger.error(f"❌ JSON parsing failed: {json_err}")
|
|
# Retry once with increased max_tokens — likely a truncation issue
|
|
if max_tokens < 16384:
|
|
logger.warning(f"Retrying with increased max_tokens ({max_tokens} → {max_tokens * 2}) due to JSON parse failure")
|
|
response_text = _retry_with_increased_tokens(
|
|
client=client,
|
|
messages=messages,
|
|
model=model,
|
|
fallback_models=fallback_models,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens * 2,
|
|
)
|
|
if response_text:
|
|
try:
|
|
parsed_json = json.loads(response_text)
|
|
if parsed_json is not None:
|
|
logger.info("✅ WaveSpeed structured JSON parsed successfully after max_tokens increase")
|
|
return parsed_json
|
|
except json.JSONDecodeError:
|
|
logger.error("❌ JSON parsing failed even after max_tokens increase")
|
|
|
|
logger.error(f"Raw response: {response_text}")
|
|
|
|
# Try to extract JSON from the response using regex
|
|
if response_text:
|
|
json_match = re.search(r'\{.*\}', response_text, re.DOTALL)
|
|
if json_match:
|
|
try:
|
|
extracted_json = json.loads(json_match.group())
|
|
logger.info("✅ JSON extracted using regex fallback")
|
|
return extracted_json
|
|
except json.JSONDecodeError:
|
|
pass
|
|
|
|
return {"error": "Failed to parse JSON response", "raw_response": response_text}
|
|
|
|
except Exception as e:
|
|
logger.error(f"❌ WaveSpeed API call failed: {e}")
|
|
# If 422 Unprocessable Entity (often due to response_format not supported), retry without it
|
|
if "422" in str(e) or "not supported" in str(e).lower() or isinstance(e, NotFoundError):
|
|
logger.info("Retrying without response_format...")
|
|
response = None
|
|
last_error = None
|
|
for candidate_model in _fallback_model_sequence(model, fallback_models):
|
|
try:
|
|
response = client.chat.completions.create(
|
|
model=candidate_model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens
|
|
)
|
|
if candidate_model != model:
|
|
logger.warning("WaveSpeed structured no-response-format fallback model: {}", candidate_model)
|
|
break
|
|
except NotFoundError as nf_err:
|
|
last_error = nf_err
|
|
logger.warning("WaveSpeed structured model not found (no response_format path): {}", candidate_model)
|
|
continue
|
|
|
|
if response is None:
|
|
raise last_error or e
|
|
response_text = response.choices[0].message.content
|
|
response_text = response_text.strip() if response_text else ""
|
|
# Parse JSON with robust cleaning
|
|
if response_text.startswith("```json"):
|
|
response_text = response_text[7:]
|
|
if response_text.startswith("```"):
|
|
response_text = response_text[3:]
|
|
if response_text.endswith("```"):
|
|
response_text = response_text[:-3]
|
|
response_text = response_text.strip()
|
|
try:
|
|
return json.loads(response_text) if response_text else {"error": "Empty response"}
|
|
except json.JSONDecodeError:
|
|
json_match = re.search(r'\{.*\}', response_text, re.DOTALL)
|
|
if json_match:
|
|
try:
|
|
return json.loads(json_match.group())
|
|
except json.JSONDecodeError:
|
|
pass
|
|
return {"error": "Failed to parse JSON response", "raw_response": response_text}
|
|
raise e
|
|
|
|
except Exception as e:
|
|
error_msg = str(e) if str(e) else repr(e)
|
|
error_type = type(e).__name__
|
|
logger.error(f"❌ WaveSpeed structured JSON generation failed [{error_type}]: {error_msg}")
|
|
|
|
# Surface balance/quota errors as HTTPException so upstream can show user-friendly messages
|
|
from fastapi import HTTPException
|
|
if "balance_not_enough" in error_msg or "403" in error_msg or "PermissionDenied" in error_type:
|
|
raise HTTPException(
|
|
status_code=403,
|
|
detail={
|
|
"error": "insufficient_balance",
|
|
"message": "WaveSpeed API balance is insufficient. Please top up your WaveSpeed account or switch to a different provider.",
|
|
"usage_info": {
|
|
"error_type": "insufficient_balance",
|
|
"provider": "wavespeed",
|
|
"suggestion": "Set GPT_PROVIDER=google in your environment to use Gemini instead, or add credits to your WaveSpeed account."
|
|
}
|
|
}
|
|
)
|
|
raise Exception(f"WaveSpeed structured JSON generation failed: {error_msg}")
|