Function generate_python_package

pygen/package_groq.py:76–145

Generate a Python package structure and content using the specified LLM via Groq.

(package_name: str, package_description: str, features: List[str], client: Groq, model_name: str)

        return ""

def generate_python_package(package_name: str, package_description: str, features: List[str], client: Groq, model_name: str) -> Optional[Dict[str, str]]:
    """Generate a Python package structure and content using the specified LLM via Groq."""

    # Format the features as a bulleted list
    if features:
        features_formatted = "\n".join(f"- {feature.strip()}" for feature in features)
        features_section = f"\n\nInclude the following features:\n{features_formatted}"
    else:
        features_section = ""

    prompt = f"""
Create a Python package named `{package_name}` based on the following description:
{package_description}{features_section}

Provide the package structure and content strictly in the following format for each file:

<file_path>
<begin_content>
file_content
<end_content>

Include at least the following files in this order:
- setup.py
- README.md
- requirements.txt
- examples/example_{package_name}.py
- tests/test_{package_name}.py
- {package_name}/__init__.py
- {package_name}/main.py

Ensure there are no extra lines or spaces between the tags and the content.

**Example:**

<setup.py>
<begin_content>
from setuptools import setup, find_packages

setup(
    name='examplepkg',
    version='0.1.0',
    description='An example package',
    author='Your Name',
    packages=find_packages(),
    install_requires=[
        # Add dependencies here
    ],
    classifiers=[
        'Programming Language :: Python :: 3',
        'License :: OSI Approved :: MIT License',
    ],
)
<end_content>
"""

    try:
        response_text = generate_content_with_groq(client, model_name, prompt)
        print("=== Raw AI Response ===")