chore: use a Conversation type (#3735)
This commit is contained in:
@@ -4,7 +4,8 @@ use dotenvy::dotenv;
|
||||
use futures::StreamExt;
|
||||
use goose::agents::{Agent, AgentEvent, ExtensionConfig};
|
||||
use goose::config::{DEFAULT_EXTENSION_DESCRIPTION, DEFAULT_EXTENSION_TIMEOUT};
|
||||
use goose::message::Message;
|
||||
use goose::conversation::message::Message;
|
||||
use goose::conversation::Conversation;
|
||||
use goose::providers::databricks::DatabricksProvider;
|
||||
|
||||
#[tokio::main]
|
||||
@@ -32,10 +33,11 @@ async fn main() {
|
||||
println!(" {}", extension);
|
||||
}
|
||||
|
||||
let messages = vec![Message::user()
|
||||
.with_text("can you summarize the readme.md in this dir using just a haiku?")];
|
||||
let conversation = Conversation::new(vec![Message::user()
|
||||
.with_text("can you summarize the readme.md in this dir using just a haiku?")])
|
||||
.unwrap();
|
||||
|
||||
let mut stream = agent.reply(&messages, None, None).await.unwrap();
|
||||
let mut stream = agent.reply(conversation, None, None).await.unwrap();
|
||||
while let Some(Ok(AgentEvent::Message(message))) = stream.next().await {
|
||||
println!("{}", serde_json::to_string_pretty(&message).unwrap());
|
||||
println!("\n");
|
||||
|
||||
@@ -1,11 +1,9 @@
|
||||
use anyhow::Result;
|
||||
use dotenvy::dotenv;
|
||||
use goose::{
|
||||
message::Message,
|
||||
providers::{
|
||||
base::{Provider, Usage},
|
||||
databricks::DatabricksProvider,
|
||||
},
|
||||
use goose::conversation::message::Message;
|
||||
use goose::providers::{
|
||||
base::{Provider, Usage},
|
||||
databricks::DatabricksProvider,
|
||||
};
|
||||
use tokio_stream::StreamExt;
|
||||
|
||||
|
||||
@@ -1,9 +1,9 @@
|
||||
use anyhow::Result;
|
||||
use base64::{engine::general_purpose::STANDARD as BASE64, Engine as _};
|
||||
use dotenvy::dotenv;
|
||||
use goose::{
|
||||
message::Message,
|
||||
providers::{bedrock::BedrockProvider, databricks::DatabricksProvider, openai::OpenAiProvider},
|
||||
use goose::conversation::message::Message;
|
||||
use goose::providers::{
|
||||
bedrock::BedrockProvider, databricks::DatabricksProvider, openai::OpenAiProvider,
|
||||
};
|
||||
use mcp_core::tool::ToolCall;
|
||||
use rmcp::model::{Content, Tool};
|
||||
|
||||
Reference in New Issue
Block a user