百度文心一言测试版接口分析

JHC000000

已于 2023-08-28 18:35:40 修改

阅读量642

点赞数

分类专栏：爬虫文章标签： python

于 2023-08-22 11:15:51 首次发布

本文链接：https://blog.youkuaiyun.com/CXY00000/article/details/132423811

版权

爬虫专栏收录该内容

40 篇文章

订阅专栏

本文介绍了如何使用Python编写一个类BDWX，通过requests库调用文心一言的API接口，解析返回的数据，支持文本和图片内容的获取。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# !/usr/bin/env python3
# -*- coding: UTF-8 -*-
"""
@author  : v_jiaohaicheng@baidu.com
@des     :

"""
import requests
import json
import re


class BDWX():
    """
    文心一言接口解析
    """
    def __init__(self):
        self.headers = {
            "Content-Type": "application/json",
            "Referer": "https://yiyan.baidu.com/",
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36",
        }
        # 换成自己的 BDUSS
        self.cookies = {
            "BDUSS": "I3TTZCM3gxLTVzZm44dmlnLWN-Q2ZQNmRScm5MR3REcENERVFrZEc0Wmlwd3RsRVFBQUFBJCQAAAAAAAAAAAEAAAAQdrtY0fS54rXE37nfud-537kAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAGIa5GRiGuRkZk"
        }

        self.base_url = "https://yiyan.baidu.com/eb/chat/conversation/v2"

    def get_data(self, query):
        """
        请求数据
        :param query:
        :return:
        """
        data = {
            "text": "{}".format(query),
            "timestamp": 1692670879294,
        }
        data = json.dumps(data, separators=(',', ':'))
        response = requests.post(
            self.base_url,
            headers=self.headers,
            cookies=self.cookies,
            data=data)
        return response.text

    def anlise_data(self, data,query):
        """
        解析数据
        :param data:
        :return:
        """
        res = re.findall("event:message\ndata:(.*)", data)
        for i in res:
            content = json.loads(i)["data"]["content"]
            if query.startswith("画"):
                img_url = re.findall('<img src="(.*)" /><br>', content)[0]
                yield img_url
            else:
                yield content


    def process(self,query):
        """
        处理逻辑 没加异常捕获（正常情况基本不会有异常）
        :param query:
        :return:
        """
        data = self.get_data(query)
        for content in self.anlise_data(data,query):
            print(content,end="")



if __name__ == '__main__':
    bdwx = BDWX()
    while True:
        bdwx.process("你能做什么")
        print()