vinzenz/hyperhive
vinzenz/hyperhive
Watch
0
0
Fork
You've already forked hyperhive
 0
Code Activity

move the review classifier, agent login and inference types into swarm-wire

Browse source
This commit is contained in:
atlas 2026-10-18 08:58:13 +02:00 • committed by mara
commit 6ea13c02e3
1 changed files with 344 additions and 332 deletions
Download patch file Download diff file Expand all files Collapse all files

14
swarm-controller/src/review_classifier.rs
View file

@ -31,7 +31,7 @@ use axum::extract::{Path, Query, State};
use axum::http::{HeaderMap, StatusCode};
use problem_details::ProblemDetails;
use serde::{Deserialize, Serialize};
use swarm_queue_client::review_classifier::{
use swarm_wire::review_classifier::{
Classification, MAX_CLASSIFICATIONS, Overall, RUN_DEADLINE, Reply, Row, SUBJECT, TryOut,
};
use utoipa::{IntoParams, ToSchema};
@ -763,7 +763,7 @@ pub(crate) mod tests {
use axum::body::Body;
use axum::http::{Request, StatusCode};
use forgejo_api::{ApiErrorKind, ForgejoError};
use swarm_queue_client::review_classifier::{Overall, Polarity, Reply, Row, Verdict};
use swarm_wire::review_classifier::{Overall, Polarity, Reply, Row, Verdict};
use tower::ServiceExt as _;
use super::*;
@ -988,13 +988,13 @@ pub(crate) mod tests {
#[tokio::test]
async fn a_try_out_runs_on_the_stored_runtime_model_and_effort() {
use swarm_queue_client::agent_inference::Effort;
use swarm_queue_client::review_classifier::Runtime;
use swarm_wire::agent_inference::Effort;
use swarm_wire::review_classifier::Runtime;
let acp = swarm_queue_client::review_classifier::Acp {
let acp = swarm_wire::review_classifier::Acp {
base_url: "https://llm.example/v1".to_owned(),
model: "qwen".to_owned(),
prices: swarm_queue_client::review_classifier::Prices {
prices: swarm_wire::review_classifier::Prices {
input: 1.0,
output: 2.0,
cache_read: None,
@ -1167,7 +1167,7 @@ pub(crate) mod tests {
let mut requests = classifier
.queue_subscribe(
SUBJECT,
swarm_queue_client::review_classifier::QUEUE_GROUP.to_owned(),
swarm_wire::review_classifier::QUEUE_GROUP.to_owned(),
)
.await
.expect("subscribe");