书生大模型实战营第4期 - 入门关卡第2关

原创于 2024-11-19 22:47:00 发布 · 286 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #ai

关卡名称

Python 基础知识

关卡任务

Leetcode 383
Vscode连接InternStudio debug笔记

作业：

1.Leetcode 383

2. Vscode连接InternStudio debug

定位错误位置

debug找错误原因

传给json解析器的内容不是标准的json字符串

修改

运行结果

完整代码

from openai import OpenAI
import json
def internlm_gen(prompt,client):
    '''
    LLM生成函数
    Param prompt: prompt string
    Param client: OpenAI client 
    '''
    response = client.chat.completions.create(
        model="internlm2.5-latest",
        messages=[
            {"role": "user", "content": prompt},
      ],
        stream=False
    )
    return response.choices[0].message.content

api_key = '----'
client = OpenAI(base_url="https://internlm-chat.intern-ai.org.cn/puyu/api/v1/",api_key=api_key)

content = """
书生浦语InternLM2.5是上海人工智能实验室于2024年7月推出的新一代大语言模型，提供1.8B、7B和20B三种参数版本，以适应不同需求。
该模型在复杂场景下的推理能力得到全面增强，支持1M超长上下文，能自主进行互联网搜索并整合信息。
"""
prompt = f"""
请帮我从以下``内的这段模型介绍文字中提取关于该模型的信息，要求包含模型名字、开发机构、提供参数版本、上下文长度四个内容，以json格式返回。
`{content}`
"""
res = internlm_gen(prompt,client)
res_json = json.loads(res.strip("```json\n").strip("\n```"))
print(res_json)