Function generate_python_package

pygen/package_ollama.py:82–170

Generate a Python package structure and content using the Ollama model.

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

    return response

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

    # 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 = ""

    # Updated prompt with clearer instructions
    prompt = f"""
You are to generate the structure and content of 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 **only** the files specified below, in the given 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.

**Important Instructions:**

- Do **not** include any text outside the specified format.
- Do **not** provide any explanations, introductions, or conclusions.
- Ensure that every `<begin_content>` has a matching `<end_content>`.
- Use the exact file paths and filenames as specified.
- The content inside `<begin_content>` and `<end_content>` should be the code or text for that file.

**Example:**

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

setup(
    name='{package_name}',
    version='0.1.0',
    description='An example package',
    author='Your Name',
    packages=find_packages(),
    install_requires=[
        # Add dependencies here
    ],
    classifiers=[