use oris_runtime::{
chain::{Chain, LLMChainBuilder},
llm::openai::{OpenAI, OpenAIModel},
prompt::HumanMessagePromptTemplate,
prompt_args, template_jinja2,
};
use std::io::{self, Write};
#[tokio::main]
async fn main() {
let prompt = HumanMessagePromptTemplate::new(template_jinja2!(
"Give me a creative name for a store that sells: {{producto}}",
"producto"
));
let llm = OpenAI::default().with_model(OpenAIModel::Gpt35);
let chain = LLMChainBuilder::new()
.prompt(prompt)
.llm(llm)
.build()
.unwrap();
print!("Please enter a product: ");
io::stdout().flush().unwrap();
let mut product = String::new();
io::stdin().read_line(&mut product).unwrap();
let product = product.trim();
let output = chain
.invoke(prompt_args!["producto" => product]) .await
.unwrap();
println!("Output: {}", output);
}