当前位置：首页 > news >正文

网站丢失怎么解决营销伎巧第一季

news 2025/7/23 10:21:22

网站丢失怎么解决,营销伎巧第一季,企业网站选wordpress和织梦,非常酷的wordpress主题文章目录前言环境准备依赖代码运行使用最后前言你好，我是醉墨居士，最近准备花一些时间来使用Rust语言实现一个智能助手，希望能够帮助到你。环境准备安装Rust语言环境，你可以从官网下载安装包安装。安装Ollama，…

文章目录

前言
环境准备
依赖
代码
运行
使用
最后

前言

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

环境准备

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

依赖

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

代码

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()})
}