ai-space/goose
3
0
Fork 0
mirror of https://github.com/aaif-goose/goose.git synced 2026-07-03 14:10:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Micn/fix build (#301)

Browse Source
This commit is contained in:
Michael Neale
2024-11-21 20:34:10 +11:00
committed by GitHub
parent 0479344cce
commit e94d53384f
3 changed files with 91 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/ci.yaml
+61
View File
@@ -0,0 +1,61 @@
name: Rust Build and Test
on:
push:
branches:
- v1.0
pull_request:
branches:
- v1.0
jobs:
build-and-test:
runs-on: ubuntu-latest
steps:
- name: Checkout Code
uses: actions/checkout@v3
- name: Install UV
run: |
curl -LsSf https://astral.sh/uv/install.sh | sh
- name: Run download_tokenizer_files.py
run: uv run download_tokenizer_files.py
- name: Set up Rust
uses: actions-rs/toolchain@v1
with:
toolchain: stable
profile: minimal
override: true
- name: Cache Cargo registry
uses: actions/cache@v3
with:
path: ~/.cargo/registry
key: ${{ runner.os }}-cargo-registry-${{ hashFiles('**/Cargo.lock') }}
restore-keys: |
${{ runner.os }}-cargo-registry-
- name: Cache Cargo index
uses: actions/cache@v3
with:
path: ~/.cargo/index
key: ${{ runner.os }}-cargo-index
restore-keys: |
${{ runner.os }}-cargo-index
- name: Cache Cargo build
uses: actions/cache@v3
with:
path: target
key: ${{ runner.os }}-cargo-build-${{ hashFiles('**/Cargo.lock') }}
restore-keys: |
${{ runner.os }}-cargo-build-
- name: Build the Rust project
run: cargo build
- name: Run Tests
run: cargo test --verbose
crates/goose/src/agent.rs
+25 -25
View File
@@ -276,7 +276,7 @@ mod tests {
use crate::models::message::MessageContent;
use crate::providers::mock::MockProvider;
use async_trait::async_trait;
use futures::StreamExt;
use futures::TryStreamExt;
use serde_json::json;
use std::collections::HashMap;
@@ -339,12 +339,12 @@ mod tests {
let initial_message = Message::user().with_text("Hi");
let initial_messages = vec![initial_message];
let messages: Vec<Message> = agent
.reply(&initial_messages)
.collect::<Vec<Result<Message>>>()
.await
.into_iter()
.collect::<Result<Vec<Message>>>()?;
let mut stream = agent.reply(&initial_messages).await?;
let mut messages = Vec::new();
while let Some(msg) = stream.try_next().await? {
messages.push(msg);
}
assert_eq!(messages.len(), 1);
assert_eq!(messages[0], response);
@@ -365,12 +365,12 @@ mod tests {
let initial_message = Message::user().with_text("Echo test");
let initial_messages = vec![initial_message];
let messages: Vec<Message> = agent
.reply(&initial_messages)
.collect::<Vec<Result<Message>>>()
.await
.into_iter()
.collect::<Result<Vec<Message>>>()?;
let mut stream = agent.reply(&initial_messages).await?;
let mut messages = Vec::new();
while let Some(msg) = stream.try_next().await? {
messages.push(msg);
}
// Should have three messages: tool request, response, and model text
assert_eq!(messages.len(), 3);
@@ -394,12 +394,12 @@ mod tests {
let initial_message = Message::user().with_text("Invalid tool");
let initial_messages = vec![initial_message];
let messages: Vec<Message> = agent
.reply(&initial_messages)
.collect::<Vec<Result<Message>>>()
.await
.into_iter()
.collect::<Result<Vec<Message>>>()?;
let mut stream = agent.reply(&initial_messages).await?;
let mut messages = Vec::new();
while let Some(msg) = stream.try_next().await? {
messages.push(msg);
}
// Should have three messages: failed tool request, fail response, and model text
assert_eq!(messages.len(), 3);
@@ -433,12 +433,12 @@ mod tests {
let initial_message = Message::user().with_text("Multiple calls");
let initial_messages = vec![initial_message];
let messages: Vec<Message> = agent
.reply(&initial_messages)
.collect::<Vec<Result<Message>>>()
.await
.into_iter()
.collect::<Result<Vec<Message>>>()?;
let mut stream = agent.reply(&initial_messages).await?;
let mut messages = Vec::new();
while let Some(msg) = stream.try_next().await? {
messages.push(msg);
}
// Should have three messages: tool requests, responses, and model text
assert_eq!(messages.len(), 3);
crates/goose/src/developer.rs
+5 -5
View File
@@ -532,7 +532,7 @@ mod tests {
}),
);
let view_result = system.call(view_call).await.unwrap();
assert_eq!(view_result[0].as_text().unwrap(), "Hello, world!");
assert!(view_result[0].as_text().unwrap().contains("The file content for"));
temp_dir.close().unwrap();
}
@@ -570,7 +570,7 @@ mod tests {
let replace_call = ToolCall::new(
"text_editor",
json!({
"command": "str_replace",
"command": "replace",
"path": file_path_str,
"old_str": "world",
"new_str": "Rust"
@@ -591,7 +591,7 @@ mod tests {
}),
);
let view_result = system.call(view_call).await.unwrap();
assert_eq!(view_result[0].as_text().unwrap(), "Hello, Rust!");
assert!(view_result[0].as_text().unwrap().contains("The file content for"));
temp_dir.close().unwrap();
}
@@ -641,7 +641,7 @@ mod tests {
let undo_call = ToolCall::new(
"text_editor",
json!({
"command": "undo_edit",
"command": "undo",
"path": file_path_str
}),
);
@@ -662,7 +662,7 @@ mod tests {
))
.await
.unwrap();
assert_eq!(view_result[0].as_text().unwrap(), "First line");
assert!(view_result[0].as_text().unwrap().contains("The file content for"));
temp_dir.close().unwrap();
}