Function enhance_package_description

app.py:110–146

Enhance the package description using the Groq model with retry logic. Args: client (Groq): The Groq client instance. model_name (str): The model name to use. package_description (str): Original package description. max_retries (int): Maximum number of retri

(client: Groq, model_name: str, package_description: str, max_retries: int = 5)

    return feature_descriptions

def enhance_package_description(client: Groq, model_name: str, package_description: str, max_retries: int = 5) -> str:
    """
    Enhance the package description using the Groq model with retry logic.

    Args:
        client (Groq): The Groq client instance.
        model_name (str): The model name to use.
        package_description (str): Original package description.
        max_retries (int): Maximum number of retries for the request.

    Returns:
        str: Enhanced package description.
    """
    prompt = f"Enhance the following package description to make it more comprehensive and user-friendly, focusing on implementation details and including elaborate explanations to help with code generation.\n\n{package_description}\n\nEnhanced Description:"
    retries = 0
    while retries < max_retries:
        try:
            response = client.chat.completions.create(
                messages=[
                    {
                        "role": "user",
                        "content": prompt,
                    }
                ],
                model=model_name,
            )
            enhanced_description = response.choices[0].message.content.strip()
            return enhanced_description
        except Exception as e:
            retries += 1
            wait_time = 2 ** retries
            print(f"Error enhancing package description: {str(e)}")
            print(f"Retrying in {wait_time} seconds... (Attempt {retries}/{max_retries})")
            time.sleep(wait_time)
    else:
        print(f"Failed to enhance package description after {max_retries} attempts.")
        return package_description

def generate_python_package(package_name: str, package_description: str, features: Optional[List[str]], api_provider: str, api_key: str, model_name: str) -> Optional[Dict[str, str]]:
    """Generate a Python package structure and content using Groq or GEMINI models."""