Rust实现智能助手 - 项目初始化

文章目录

前言

你好,我是醉墨居士,最近准备花一些时间来使用Rust语言实现一个智能助手,希望能够帮助到你。

环境准备

  1. 安装Rust语言环境,你可以从官网下载安装包安装。
  2. 安装Ollama,你可以在官网下载安装包安装。(我这里暂时qwen2.5:0.5b模型)
  3. 安装PostgreSQL数据库。

依赖

toml 复制代码
ollama-rs = "0.2.2"
tokio = { version = "1.43.0", features = ["full"] }
tokio-stream = { version = "0.1.17" }
axum = "0.8.1"

代码

rust 复制代码
use std::{collections::HashMap, sync::OnceLock};

use axum::{
    extract::Query, routing::get, Json, Router
};
use ollama_rs::{
    generation::completion::request::GenerationRequest, Ollama
};

#[derive(Debug)]
struct Context {
    ollama: Ollama,
}

static CONTEXT: OnceLock<Context> = OnceLock::new();

impl Context {
    fn global() -> &'static Self {
        CONTEXT.get_or_init(|| {
            println!("Initializing Ollama...");
            Context {
                ollama: Ollama::default(),
            }
        })
    }
}

#[tokio::main]
async fn main() {
    let app = Router::new().route("/", get(chat));

    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
    axum::serve(listener, app).await.unwrap();
}

async fn chat(Query(mut query): Query<HashMap<String, String>>) -> Json<String> {
    let prompt = if let Some(prompt) = query.remove("prompt") {
        prompt
    } else {
        return Json("Error: prompt not provided".to_string());
    };

    let req = GenerationRequest::new("qwen2.5:0.5b".into(), prompt);
    let res = Context::global().ollama.generate(req).await;
    Json(
        if let Ok(res) = res {
            res.response
        } else {
            "Error".to_string()
        }
    )
}

运行

bash 复制代码
cargo run

使用

打开浏览器,访问http://localhost:3000?prompt=你好,即可得到AI回复

最后

我是醉墨居士,这篇博客就先到这里,后续我会继续更新,欢迎你提出宝贵意见,互相交流,共同进步

相关推荐
海绵宝宝汉堡包1 小时前
c# 项目 文件夹
开发语言·c#
小白要加油努力2 小时前
C++设计模式--策略模式与观察者模式
开发语言·c++·设计模式
你的人类朋友2 小时前
【Node&Vue】什么是ECMAScript?
前端·javascript·后端
小马学嵌入式~2 小时前
数据结构:队列 二叉树
c语言·开发语言·数据结构·算法
你的人类朋友3 小时前
说说你对go的认识
后端·云原生·go
我崽不熬夜3 小时前
Java中基本的输入输出(I/O)操作:你知道如何处理文件吗?
java·后端·java ee
我崽不熬夜3 小时前
Java的异常处理机制:如何优雅地捕获和抛出异常?
java·后端·java ee
Slaughter信仰3 小时前
深入理解Java虚拟机:JVM高级特性与最佳实践(第3版)第二章知识点问答(21题)
java·开发语言·jvm
Ice__Cai3 小时前
Flask 之 Cookie & Session 详解:用户状态管理
后端·python·flask·cookie·session
焊锡与代码齐飞4 小时前
嵌入式第三十五课!!Linux下的网络编程
linux·运维·服务器·开发语言·网络·学习·算法