openai_orch/
utils.rs

1//! Utilites for use when writing custom requests.
2
3use async_openai::{config::OpenAIConfig, Client as OpenAIClient};
4
5use crate::keys::Keys;
6
7pub(crate) fn get_openai_client(keys: &Keys) -> OpenAIClient<OpenAIConfig> {
8  let config = OpenAIConfig::new().with_api_key(&keys.openai_api_key);
9  let config = match &keys.openai_org_id {
10    Some(openai_org_id) => config.with_org_id(openai_org_id),
11    None => config,
12  };
13  OpenAIClient::<OpenAIConfig>::with_config(config)
14}