rustglm 1.0.0

Production-oriented async Rust SDK for Zhipu GLM, Realtime, Batch, RAG, and MCP
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use futures_util::StreamExt;
use rustglm::{
    FunctionDefinition, Glm52, ResponseContent, Tool, ToolStreamEvent, TypedChatRequest,
    ZhipuClient,
};
use serde_json::json;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let tool = Tool::function(
        FunctionDefinition::new(
            "deployment_status",
            json!({
                "type": "object",
                "properties": {"service": {"type": "string"}},
                "required": ["service"]
            }),
        )
        .description("Read the current deployment status"),
    );
    let request = TypedChatRequest::<Glm52>::new()
        .tool(tool)
        .tool_stream()
        .user("Check the deployment status of payments-api.");
    let client = ZhipuClient::new(std::env::var("ZHIPU_API_KEY")?)?;
    let mut stream = client.typed_chat_tool_stream(&request).await?;
    while let Some(event) = stream.next().await {
        match event? {
            ToolStreamEvent::ContentDelta {
                content: ResponseContent::Text(text),
                ..
            } => print!("{text}"),
            ToolStreamEvent::ToolCallCompleted(call) => {
                println!("tool={} arguments={}", call.name, call.arguments);
            }
            _ => {}
        }
    }
    Ok(())
}