dirge-agent 0.12.0

Minimalistic coding agent written in Rust, optimized for memory footprint and performance
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

//! One place the provider list lives for stream-fn construction
//! (dirge-iy20).
//!
//! `AnyAgent::build_stream_fn_with_filter` (matching `AnyAgentInner`)
//! and `AnyModel::build_stream_fn` (matching `AnyModel`) both
//! enumerate the same eight providers and call the same
//! `rig_stream_fn_from_model_with_filter` helper. They were two
//! parallel 8-arm matches: adding a provider meant editing both, and
//! the compiler couldn't catch a missed arm (each match was already
//! exhaustive over its own enum). This macro is the single list.

/// Dispatch over a provider enum to build a `StreamFn`.
///
/// `$value` is matched against `$enum::{OpenRouter,…,Custom}`. Each
/// arm binds `$bind` and evaluates `$model` (written in terms of
/// `$bind`) to get the model to stream from. `tools`/`timeout`/
/// `provider`/`filter` are pasted into every arm — match arms are
/// mutually exclusive, so a moved value (e.g. `tools` without a
/// clone) is fine.
macro_rules! dispatch_stream_fn {
    (
        match $value:expr ;
        $enum:ident ( $bind:ident ) => $model:expr ,
        tools = $tools:expr ,
        timeout = $timeout:expr ,
        provider = $provider:expr ,
        filter = $filter:expr $(,)?
    ) => {{
        use $crate::agent::agent_loop::rig_stream_fn_from_model_with_filter as __stream_fn;
        // dirge-480 follow-up: OpenAI's Responses API needs the per-message
        // conversion (drop historical reasoning, synthesize tool call_ids) keyed
        // on the canonical "openai" provider identity. All three OpenAI client
        // variants — API key, ChatGPT-OAuth, and Codex — are ProviderKind::OpenAI
        // and hit that endpoint, but a custom ALIAS (provider_type = "openai"
        // named e.g. "my-gpt", or an escalation/critic alias) would otherwise
        // pass the alias string and miss the fix. The dispatch arm knows the
        // concrete variant, so force the canonical identity here regardless of
        // alias. Non-OpenAI providers keep their passed identity.
        match $value {
            $enum::OpenRouter($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
            $enum::OpenAI($bind) => __stream_fn(
                $model,
                $tools,
                $timeout,
                Some("openai".to_string()),
                $filter,
            ),
            $enum::ChatGptOpenAI($bind) => __stream_fn(
                $model,
                $tools,
                $timeout,
                Some("openai".to_string()),
                $filter,
            ),
            $enum::OpenAICodex($bind) => __stream_fn(
                $model,
                $tools,
                $timeout,
                Some("openai".to_string()),
                $filter,
            ),
            $enum::Anthropic($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
            $enum::AnthropicOauth($bind) => {
                __stream_fn($model, $tools, $timeout, $provider, $filter)
            }
            $enum::Gemini($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
            $enum::DeepSeek($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
            $enum::Glm($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
            $enum::Ollama($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
            $enum::Custom($bind) => __stream_fn($model, $tools, $timeout, $provider, $filter),
        }
    }};
}

pub(crate) use dispatch_stream_fn;