代码生成

代码

cookbook/models/vllm/code_generation.py

from agno.agent import Agent
from agno.models.vllm import vLLM

agent = Agent(
    model=vLLM(id="deepseek-ai/deepseek-coder-6.7b-instruct"),
    description="You are an expert Python developer.",
    markdown=True,
)

agent.print_response(
    """编写一个 Python 函数，使用动态规划返回第 n 个斐波那契数。"""
)

用法

1. 创建虚拟环境

   打开 Terminal 并创建一个 python 虚拟环境。

   Mac

   python3 -m venv .venv
   source .venv/bin/activate
   
2. 安装库

   pip install -U agno openai vllm
   
3. 启动 vLLM 服务

   vllm serve deepseek-ai/deepseek-coder-6.7b-instruct \
       --dtype float32 \
       --tool-call-parser pythonic
   
4. 运行 Agent

   python cookbook/models/vllm/code_generation.py