最近两年,大模型彻底火了。

从 ChatGPT、Claude 到国内的通义、文心一言,AI 正在快速改变软件行业。

很多程序员都会问一个问题:

我能不能做一个自己的 ChatGPT?

答案是:完全可以,而且比你想象的简单得多。

事实上,用 Python 只需要几十行代码,就能做出一个属于自己的 AI 聊天机器人。

今天这篇文章,我们就用 Python + 大模型 API,一步一步搭建一个 自己的 ChatGPT。

整个过程只需要 差不多10分钟。

一、最终效果

我们先看看最终效果。

运行程序后,你就可以和 AI 进行对话:

你:Python是什么

AI:Python是一种高级编程语言,具有简单易读的语法...

或者

你:写一个快速排序

AI:以下是Python实现的快速排序代码...

也就是说,你可以拥有一个:

  • AI编程助手

  • AI问答机器人

  • AI学习助手

而这一切,只需要 Python。

二、准备环境

首先需要准备 Python 环境。

推荐版本:Python 3.9+

然后安装需要的库:pip install openai

如果你打算做网页版本,还可以安装:pip install streamlit

三、调用大模型 API

接下来,我们使用 Python 调用 AI 模型。

创建一个py文件,简单实现:

from openai import OpenAI

client = OpenAI(
    api_key="你的API_KEY"
)

response = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[
        {"role": "user", "content": "介绍一下Python"}
    ]
)

print(response.choices[0].message.content)

AI 就会返回结果。

这其实就是 ChatGPT 的基本原理:

用户输入

发送给AI模型

AI生成回答

返回结果

四、实现聊天机

上面的代码只能问一次问题。

现在我们把它升级成 可以连续聊天的版本。

代码如下:

from openai import OpenAI

client = OpenAI(api_key="你的API_KEY")

messages = []

while True:

    question = input("你:")

    messages.append({
        "role": "user",
        "content": question
    })

    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=messages
    )

    answer = response.choices[0].message.content

    print("AI:", answer)

    messages.append({
        "role": "assistant",
        "content": answer
    })

运行后,你就拥有一个 终端版 ChatGPT。

示例:

你:Python是什么

AI:Python是一种高级编程语言... 你:Python能做什么 AI:Python可以用于Web开发、AI开发...

这就是一个 最基础的 AI 聊天机器人。

五、做一个 AI 网页应用

如果只是终端聊天,体验还不够好。

我们可以用 Streamlit 做一个 Web 页面。

创建文件app.py:

import streamlit as st
from openai import OpenAI

client = OpenAI(api_key="你的API_KEY")

st.title("我的AI助手")

question = st.text_input("请输入问题")

if question:

    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "user", "content": question}
        ]
    )

    st.write(response.choices[0].message.content)

运行:

streamlit run app.py

浏览器会自动打开。你就拥有了一个 自己的 AI 网页助手。

六、可以扩展哪些功能?

其实到这里,你已经拥有了一个基础版 ChatGPT。

但还可以继续升级。

例如:

1. AI知识库

让 AI 读取你的文档,例如:

  • PDF

  • Word

  • Markdown

然后回答问题。

技术:

  • LangChain

  • LlamaIndex

2. AI编程助手

可以做一个:

  • 自动写代码

  • 自动解释代码

  • 自动生成脚本

类似 Copilot。

3. AI客服系统

很多公司已经在做AI客服机器人

可以回答:

  • 产品问题

  • 使用问题

  • 常见问题

4. AI Agent

未来最火的方向是:AI Agent

AI 不只是聊天,而是:

  • 自动查资料

  • 自动写报告

  • 自动执行任务

常见框架:

  • CrewAI

  • LangGraph

  • AutoGen

七、总结

如果你想进入 AI开发领域,其实并没有那么难。

今天这篇文章,我们只用了:

  • Python

  • 一个 AI API

  • 几十行代码

就实现了一个属于自己的 ChatGPT。

学习路线可以这样走:

1️⃣ Python基础 2️⃣ 调用大模型API 3️⃣ LangChain / LlamaIndex 4️⃣ AI Agent开发

未来的软件开发,很可能会变成:

AI + 编程

而 Python,依然是 AI 世界最重要的语言。

八、最后

如果你对 Python / AI / 大模型开发 感兴趣,

后面我还会继续分享:

  • Python AI 实战项目

  • AI Agent 开发教程

  • AI 自动化工具

欢迎关注公众号“IT小知识大分享”,一起学习 AI 开发 🚀

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐