快捷導(dǎo)航

Python實現(xiàn)Ollama的提示詞生成與優(yōu)化

更新時間：2024年12月11日 15:54:58 作者：老大白菜

這篇文章主要為大家詳細介紹了Python實現(xiàn)Ollama的提示詞生成與優(yōu)化的相關(guān)知識,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

1. 基礎(chǔ)環(huán)境配置

import requests
import json
from typing import List, Dict, Optional
from dataclasses import dataclass

@dataclass
class PromptContext:
    task: str
    domain: str
    requirements: List[str]

class OllamaService:
    def __init__(self, base_url: str = "http://localhost:11434"):
        self.base_url = base_url
        self.models = {
            'mistral': 'mistral',
            'llama2': 'llama2',
            'neural-chat': 'neural-chat'
        }

2. 核心功能實現(xiàn)

2.1 提示詞生成服務(wù)

class PromptGenerationService:
    def __init__(self, model_name: str = 'mistral'):
        self.model_name = model_name
        self.api_url = "http://localhost:11434/api/generate"

    async def generate_prompt(self, context: PromptContext) -> str:
        prompt = f"""
        Task: Create a detailed prompt for the following context:
        - Task Type: {context.task}
        - Domain: {context.domain}
        - Requirements: {', '.join(context.requirements)}
        
        Generate a structured prompt that includes:
        1. Context setting
        2. Specific requirements
        3. Output format
        4. Constraints
        5. Examples (if applicable)
        """

        response = requests.post(
            self.api_url,
            json={
                "model": self.model_name,
                "prompt": prompt,
                "stream": False
            }
        )
        
        return response.json()["response"]

    async def optimize_prompt(self, original_prompt: str) -> Dict:
        prompt = f"""
        Analyze and optimize the following prompt:
        "{original_prompt}"
        
        Provide:
        1. Improved version
        2. Explanation of changes
        3. Potential variations
        """

        response = requests.post(
            self.api_url,
            json={
                "model": self.model_name,
                "prompt": prompt,
                "stream": False
            }
        )
        
        return response.json()["response"]

2.2 提示詞模板管理

class PromptTemplates:
    @staticmethod
    def get_code_review_template(code: str) -> str:
        return f"""
        Analyze the following code:
        [code]
        
        Provide:
        1. Code quality assessment
        2. Potential improvements
        3. Security concerns
        4. Performance optimization
        """

    @staticmethod
    def get_documentation_template(component: str) -> str:
        return f"""
        Generate documentation for:
        {component}
        
        Include:
        1. Overview
        2. API reference
        3. Usage examples
        4. Best practices
        """

    @staticmethod
    def get_refactoring_template(code: str) -> str:
        return f"""
        Suggest refactoring for:
        [code]
        
        Consider:
        1. Design patterns
        2. Clean code principles
        3. Performance impact
        4. Maintainability
        """

3. 使用示例

async def main():
    # 初始化服務(wù)
    prompt_service = PromptGenerationService(model_name='mistral')
    
    # 代碼生成提示詞示例
    code_context = PromptContext(
        task='code_generation',
        domain='web_development',
        requirements=[
            'React component',
            'TypeScript',
            'Material UI',
            'Form handling'
        ]
    )
    
    code_prompt = await prompt_service.generate_prompt(code_context)
    print("代碼生成提示詞：", code_prompt)
    
    # 文檔生成提示詞示例
    doc_context = PromptContext(
        task='documentation',
        domain='API_reference',
        requirements=[
            'OpenAPI format',
            'Examples included',
            'Error handling',
            'Authentication details'
        ]
    )
    
    doc_prompt = await prompt_service.generate_prompt(doc_context)
    print("文檔生成提示詞：", doc_prompt)

    # 提示詞優(yōu)化示例
    original_prompt = "寫一個React組件"
    optimized_prompt = await prompt_service.optimize_prompt(original_prompt)
    print("優(yōu)化后的提示詞：", optimized_prompt)

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())

4. 工具類實現(xiàn)

class PromptUtils:
    @staticmethod
    def format_requirements(requirements: List[str]) -> str:
        return "\n".join([f"- {req}" for req in requirements])

    @staticmethod
    def validate_prompt(prompt: str) -> bool:
        # 簡單的提示詞驗證
        return len(prompt.strip()) > 0

    @staticmethod
    def enhance_prompt(prompt: str) -> str:
        # 添加通用的提示詞增強
        return f"""
        {prompt}
        
        Additional requirements:
        - Provide clear and detailed explanations
        - Include practical examples
        - Consider edge cases
        - Follow best practices
        """

5. 錯誤處理

class PromptGenerationError(Exception):
    pass

class ModelConnectionError(Exception):
    pass

def handle_api_errors(func):
    async def wrapper(*args, **kwargs):
        try:
            return await func(*args, **kwargs)
        except requests.exceptions.ConnectionError:
            raise ModelConnectionError("無法連接到Ollama服務(wù)")
        except Exception as e:
            raise PromptGenerationError(f"提示詞生成錯誤: {str(e)}")
    return wrapper

6. 配置管理

class Config:
    MODELS = {
        'mistral': {
            'name': 'mistral',
            'description': '快速、輕量級提示詞生成',
            'parameters': {
                'temperature': 0.7,
                'max_tokens': 2000
            }
        },
        'llama2': {
            'name': 'llama2',
            'description': '復(fù)雜、詳細的提示詞需求',
            'parameters': {
                'temperature': 0.8,
                'max_tokens': 4000
            }
        },
        'neural-chat': {
            'name': 'neural-chat',
            'description': '交互式提示詞優(yōu)化',
            'parameters': {
                'temperature': 0.9,
                'max_tokens': 3000
            }
        }
    }

使用這個Python實現(xiàn)，你可以：