aiapi

use aiapi::{role, Api};
use aok::{Result, OK};
use static_init::constructor;
// use tran_ext::{parse, Ext};

#[constructor(0)]
extern "C" fn init() {
  loginit::init()
}

fn defrag(input: String) -> Result<String> {
  // let md = parse(Ext::Md, &input)?;
  Ok(
    input
      .split("\n")
      .enumerate()
      .map(|(n, p)| format!("{} {}", n + 1, p))
      // .map(|(n, p)| format!("{} {}", n + 1, md.txt_li[p]))
      .collect::<Vec<_>>()
      .join("\n"),
  )
}

#[tokio::test]
async fn test() -> Result<()> {
  // let token_li: Vec<String> = std::env::var("DOUBAO")
  // let token_li: Vec<String> = std::env::var("BIGMODEL")

  let token_li = genv::li("SILICONFLOW");
  let ai = aiapi::site::siliconflow::new(token_li);
  // let ai = aiapi::site::doubao::new(token_li);
  // let ai = aiapi::site::bigmodel::new(token_li);
  let to_name = "中文";
  let to_txt = r#"《IEEE 软件工程学报》今年 6 月刊就发表了一项研究，从功能性、复杂性与安全性等方面评估了 OpenAI 的 ChatGPT 所生成的代码。结果表明，ChatGPT 在生成功能性代码方面取得了极大的成功，成功率最低为 0.66% ，最高可达 89% ，具体要取决于任务的难度、编程语言等许多其他因素。
虽然在某些情况下，人工智能生成器生成的代码比人类写的还要好，但分析也揭示了人工智能生成的代码存在一些安全问题。"#;
  let to_txt = defrag(to_txt.into())?;
  println!("{}", to_txt);
  let r = ai
        .send([
          (
            role::SYSTEM,
            format!(
              "你是编辑校对,请逐句微调{to_name}markdown(首列为行号),润色表达。输出保留行号,不增删换行,不添油加醋,不要破坏markdown格式和html标签。"
            ),
          ),
          (role::ASSISTANT, "请输入文章".into()),
          (role::USER, to_txt),
        ])
        .await?;
  dbg!(&r);
  let txt = r
    .choices
    .into_iter()
    .map(|c| c.message.content)
    .collect::<Vec<_>>()
    .join("\n");
  println!("{}", txt);

  // for i in aiapi::md(&ai, Lang::Zh, &[Lang::En], zh, &[en]).await? {
  //   println!("{}", i);
  // }
  OK
}
aiapi 0.1.5

aiapi
