超实用Python工具!一键调用OpenAI/Claude生成文本摘要,开箱即用

python                  import os                  from typing import Optional                  def summarize(                  text: str,                  model: str = "gpt-3.5-turbo",                  api_type: str = "openai",                  api_key: Optional[str] = None,                  max_tokens: int = 300,                  temperature: float = 0.5                  ) -> str:                  """                  调用大模型 API 对输入文本进行摘要。                  Args:                  text: 待摘要的原始文本。                  model: 模型名称，例如 'gpt-3.5-turbo'、'claude-3-haiku-20240307' 等。                  api_type: API 类型，'openai' 或 'claude'。                  api_key: API 密钥，若不提供则从环境变量读取。                  max_tokens: 生成摘要的最大 token 数。                  temperature: 生成温度（0~1），值越低结果越确定。                  Returns:                  摘要文本。                  Raises:                  ValueError: 不支持的 api_type 或缺少 API 密钥。                  Exception: API 调用失败时抛出异常。                  """                  # 1. 获取 API 密钥                  if api_key is None:                  if api_type == "openai":                  api_key = os.environ.get("OPENAI_API_KEY")                  elif api_type == "claude":                  api_key = os.environ.get("ANTHROPIC_API_KEY")                  else:                  raise ValueError("api_type 必须是 'openai' 或 'claude'")                  if not api_key:                  raise ValueError(f"未找到 {api_type.upper()} API 密钥，请通过参数或环境变量提供")                  # 2. 构建 prompt                  prompt = f"请将以下文本总结为一段简洁的摘要：\n\n{text}"                  # 3. 根据 API 类型调用                  if api_type == "openai":                  from openai import OpenAI                  client = OpenAI(api_key=api_key)                  response = client.chat.completions.create(                  model=model,                  messages=[{"role": "user", "content": prompt}],                  max_tokens=max_tokens,                  temperature=temperature,                  )                  summary = response.choices[0].message.content.strip()                  elif api_type == "claude":                  from anthropic import Anthropic                  client = Anthropic(api_key=api_key)                  response = client.messages.create(                  model=model,                  max_tokens=max_tokens,                  temperature=temperature,                  messages=[{"role": "user", "content": prompt}],                  )                  summary = response.content[0].text.strip()                  else:                  raise ValueError("api_type 必须是 'openai' 或 'claude'")                  return summary

bash                  pip install openai anthropic

python                  # 可提前配置环境变量 OPENAI_API_KEY，也可直接传入api_key参数                  result = summarize(                  text="这里粘贴需要总结的长篇文本、文章、会议纪要等内容",                  model="gpt-3.5-turbo",                  api_type="openai"                  )                  print("文本摘要：", result)

python                  # 可提前配置环境变量 ANTHROPIC_API_KEY，也可直接传入api_key参数                  result = summarize(                  text="这里粘贴需要总结的长篇文本、文章、会议纪要等内容",                  model="claude-3-haiku-20240307",                  api_type="claude"                  )                  print("文本摘要：", result)