mirror of
https://github.com/aljazceru/goose.git
synced 2025-12-18 06:34:26 +01:00
[goose-llm] add completion request & error (#2451)
This commit is contained in:
@@ -3,7 +3,9 @@ use std::vec;
|
||||
use anyhow::Result;
|
||||
use goose_llm::{
|
||||
completion,
|
||||
types::completion::{CompletionResponse, ExtensionConfig, ToolApprovalMode, ToolConfig},
|
||||
types::completion::{
|
||||
CompletionRequest, CompletionResponse, ExtensionConfig, ToolApprovalMode, ToolConfig,
|
||||
},
|
||||
Message, ModelConfig,
|
||||
};
|
||||
use serde_json::json;
|
||||
@@ -91,13 +93,13 @@ async fn main() -> Result<()> {
|
||||
println!("\n---------------\n");
|
||||
println!("User Input: {text}");
|
||||
let messages = vec![Message::user().with_text(text)];
|
||||
let completion_response: CompletionResponse = completion(
|
||||
let completion_response: CompletionResponse = completion(CompletionRequest::new(
|
||||
provider,
|
||||
model_config.clone(),
|
||||
system_preamble,
|
||||
&messages,
|
||||
&extensions,
|
||||
)
|
||||
))
|
||||
.await?;
|
||||
// Print the response
|
||||
println!("\nCompletion Response:");
|
||||
|
||||
Reference in New Issue
Block a user