Poe Bot Process

這是一個用 Rust 實現的 Poe API 客戶端庫。它允許您與 Poe API 平台進行交互，發送查詢請求並接收回應。

功能

流式接收 bot 回應
獲取可用模型列表

安裝

在您的 Cargo.toml 文件中添加以下依賴：

[dependencies]
poe_api_process = "0.1.1"

使用方法創建客戶端並發送請求

use poe_api_process::{PoeClient, QueryRequest, ProtocolMessage, EventType};
use futures_util::StreamExt;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let client = PoeClient::new("your_bot_name", "your_access_key");

    let request = QueryRequest {
        version: "1".to_string(),
        r#type: "query".to_string(),
        query: vec![ProtocolMessage {
            role: "user".to_string(),
            content: "你好，最近如何？".to_string(),
            content_type: "text/markdown".to_string(),
        }],
        user_id: "".to_string(),
        conversation_id: "".to_string(),
        message_id: "".to_string(),
    };

    let mut stream = client.stream_request(request).await?;
    let mut current_response = String::new();

    while let Some(response) = stream.next().await {
        match response {
            Ok(event) => match event.event {
                EventType::Text => {
                    if let Some(partial) = event.data {
                        if !partial.text.is_empty() {
                            println!("收到文字: {}", partial.text);
                            current_response.push_str(&partial.text);
                        }
                    }
                },
                EventType::ReplaceResponse => {
                    if let Some(partial) = event.data {
                        println!("替換回應: {}", partial.text);
                        current_response = partial.text;
                    }
                },
                EventType::Done => {
                    println!("對話完成");
                    println!("最終回應: {}", current_response);
                    break;
                },
            },
            Err(e) => eprintln!("錯誤: {}", e),
        }
    }

    Ok(())
}

獲取可用模型列表

use poe_api_process::get_model_list;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let models = get_model_list(Some("zh-Hant")).await?;
    for model in models.data {
        println!("Model ID: {}", model.id);
        println!("Description: {}", model.description);
        println!("---");
    }

    Ok(())
}

注意事項請確保您有有效的 Poe API 訪問密鑰。使用 stream_request 方法時，請提供有效的 bot 名稱和訪問密鑰。 get_model_list 函數不需要訪問密鑰，可以直接使用。

poe_api_process 0.1.1

Poe Bot Process

功能

安裝