package main

import (
    "log"
    "github.com/gonewx/contextor"
)

func main() {
    // 初始化客户端
    client, err := contextor.NewClient(`{
        "surrealdb_url": "ws://localhost:8000",
        "redis_url": "redis://localhost:6379",
        "openai_api_key": "your-api-key"
    }`)
    if err != nil {
        log.Fatal(err)
    }

    // 创建会话
    session, err := client.Session.Start("tenant-123")
    if err != nil {
        log.Fatal(err)
    }

    // 注入数据
    err = client.Memory.Ingest(session.ID(), "项目截止日期是2025年12月31日")
    if err != nil {
        log.Fatal(err)
    }

    // 检索上下文
    results, err := client.Retriever.Search(session.ID(), "截止日期是什么时候？")
    if err != nil {
        log.Fatal(err)
    }

    log.Printf("检索结果: %+v", results)
}