QQ聊天机器人OneBot部署的探索

QQ聊天机器人OneBot部署的探索

贴个垃圾代码。。
底下有写得好一点的垃圾。。。。。

import aiohttp
import asyncio
import json,jsonpath
import ollama
from ollama import Client
client = Client(host='http://localhost:11434')
from ollama import AsyncClient

async def chat(content:str) -> str:
  message = {'role': 'user', 'content': content}
  response = await AsyncClient(host="http://localhost:11434").chat(model='qwen2:7b', messages=[message])
  return response["message"]["content"]

str1 = '{\
    "action": "send_group_msg",\
    "params": {\
        "group_id": 175736385,\ # 这里的group_id指的是群聊号，user_id则为QQ好友号码
        "message": [{\
            "type": "text",\
            "data": {\
                "text": "'
str2 = '"\
            }}, {\
            "type": "image",\
            "data": {\
                "file": "https://moe.jitsu.top/img/"\
            }\
            }]\
    },\
    "echo": "123"\
}'
async def connect_websocket(url):
    try:
        async with aiohttp.ClientSession() as session:
            async with session.ws_connect(url) as ws:
                await ws.send_str(str1)

                async for msg in ws:
                    if msg.type == aiohttp.WSMsgType.TEXT:
                        print(msg.data)
                        a = json.loads(msg.data)
                        # b = jsonpath.jsonpath(a,'$..raw_message')
                        #if(type(a["data"]) == dict):
                        try:
                            
                            print("1111111111111:",a["raw_message"])
                        except KeyError as ke:
                            pass
                        try:
                            print('raw_message',a["raw_message"])
                            # response = client.chat(model='qwen2:7b', messages=[
                            #         {
                            #             'role': 'user',
                            #             'content': a["raw_message"],
                            #         },
                            #     ])
                            response = asyncio.run(await chat(a["raw_message"]))
                            #response = '啊啊啊'
                            str = str1 + response + str2
                            await ws.send_str(str)
                        except:
                            print('aaaaaaa~aaaaaaaaa')
                    elif msg.type == aiohttp.WSMsgType.CLOSED:
                        break

    except aiohttp.ClientError as err:
        print(f'Error: {err}')

loop = asyncio.get_event_loop()
loop.run_until_complete(connect_websocket('http://localhost:3001'))

下面是目前在写的，在用的，AI没有上下文就很难受了。。。。搞半天没弄出个所以然
但是只能贴一部分

• AIollama.py

import ollama
from ollama import AsyncClient
#client = ollama.Client(host="localhost:11434")
client = AsyncClient(host="http://localhost:11434")

async def chat(content:str) -> str:
    message = {'role': 'user', 'content': content}
    # context1 = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
    # response = await client.generate(model='qwen2:7b',context=context1,prompt=content,stream=False)
    # return response["response"]
    response = await client.chat(model='qwen2:7b',messages=[message],stream=False)
    return response["message"]["content"]

# def get_response(prompt) -> str:
    return client.chat(
        model="qwen2:7b",
        messages=[
            {
                "role": "user",
                "content": prompt,
            },
        ],
        stream=False
    )

• main.py

import asyncio
from re import split
import AIollama
from bot import Bot, Command
from message import MessageList
from event import Event, PostType
import requests as req
async def msgProc(bot: Bot, m: Event):
    print(m.check().raw)
    zhiling = m.check().get_msg()
    print(zhiling)
    match zhiling:
        case _ if "/echo" in zhiling and zhiling[:5] == "/echo":
            if(zhiling != "/echo"):
                zhiling = zhiling.replace("/echo ", "")
            await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text(zhiling)])
        case "/bing":
            zhiling = req.request(
                "get",
                "https://cn.bing.com/HPImageArchive.aspx",
                params={"format": "js", "idx": 0, "n": 1}
            )
            print(zhiling.text)
            zhiling = zhiling.json()
            zhiling = "https://cn.bing.com" + zhiling["images"][0]["url"]
            
            #"https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1"
            await bot.send_group_msg(m.check().raw["group_id"],[MessageList.image(str(zhiling))])
        case "/random":
            await bot.send_group_msg(m.check().raw["group_id"],[MessageList.image("https://moe.jitsu.top/img/")])
        case "/help":
            await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("/bing——必应每日一图\n/dev——发送调试信息\n/random——随机二次元\n/help——帮助\n/echo——重复话语\n/music——点歌\n/ai——与AI对话")])
        case _ if "/ai" in zhiling and zhiling[:3] == "/ai":
            chatText = zhiling.replace("/ai", "")
            await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text(await AIollama.chat(chatText))])
        case _ if "/music" in zhiling and zhiling[:6] == "/music":
            if(zhiling == "/music"):
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("点歌格式：\n/music type id\n/music 1 title url\n/music 2 title openUrl audioUrl")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("type：\n1：普通点歌\n2：普通点歌\n163：网易云点歌\nxm：虾米点歌\nqq：QQ音乐点歌")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("id：\n1：普通点歌：歌曲名\n2：普通点歌：歌曲名\n163：网易云id\nxm：虾米id\nqq：QQ音乐id")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("url：\n普通点歌：歌曲url")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("openUrl：\n普通点歌：点击打开的url")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("audioUrl：\n普通点歌：歌曲url")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("示例：\n/music 1 歌曲名 歌曲url\n/music 2 歌曲名 点击打开的url 歌曲url\n/music 163 网易云id\n/music xm 虾米id\n/music qq QQ音乐id")])
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.text("注意：\n/music 1 2 163 xm qq 后面必须跟空格")])
                await bot.send_group_msg(m.check().raw["group_id"],[{
    "type": "music",
    "data": {
        "type": "custom",
        "url": "https://lw-sycdn.kuwo.cn/87898d2bfa406816e7a24dee4434e67c/66b3a70e/resource/30106/trackmedia/M5000004J5Dg2Aifkc.mp3?bitrate$128&from=vip",
        "audio": "https://lw-sycdn.kuwo.cn/87898d2bfa406816e7a24dee4434e67c/66b3a70e/resource/30106/trackmedia/M5000004J5Dg2Aifkc.mp3?bitrate$128&from=vip",
        "title": "Aurora - Hoaprox/Dang Minh"
    }
}])
                pass
            musicNameAndUrl = zhiling.replace("/music", "").split()
            print("musicNameAndUrl",musicNameAndUrl)
            if musicNameAndUrl[0] == "163" or musicNameAndUrl[0] == "xm" or musicNameAndUrl[0] == "qq":
                await bot.send_group_msg(m.check().raw["group_id"],[MessageList.Music(musicNameAndUrl[0], musicNameAndUrl[1])])# id点歌 /music type id
            musicTitle = " ".join(musicNameAndUrl[1:len(musicNameAndUrl)-1])
            match musicNameAndUrl[0]:
                case "1":
                    await bot.send_group_msg(m.check().raw["group_id"],[MessageList.CustomMusic(url=musicNameAndUrl.pop(), audio=musicNameAndUrl.pop(),title=musicTitle)])# 普通点歌 /music 1 title Url
                case "2":
                    await bot.send_group_msg(m.check().raw["group_id"],[MessageList.CustomMusic(url=musicNameAndUrl[len(musicNameAndUrl)-2], audio=musicNameAndUrl.pop(),title=musicTitle)])# 普通点歌 /music 2 title OpenUrl AudioUrl

'''
{
    "type": "music",
    "data": {
        "type": "custom",
        "url": "https://lw-sycdn.kuwo.cn/87898d2bfa406816e7a24dee4434e67c/66b3a70e/resource/30106/trackmedia/M5000004J5Dg2Aifkc.mp3?bitrate$128&from=vip",
        "audio": "https://lw-sycdn.kuwo.cn/87898d2bfa406816e7a24dee4434e67c/66b3a70e/resource/30106/trackmedia/M5000004J5Dg2Aifkc.mp3?bitrate$128&from=vip",
        "title": "Aurora - Hoaprox/Dang Minh"
    }
}
'''

async def main():
    cmd = Command("")
    bot = Bot("http://localhost:3001", cmd)
    cmd.cmds.append(msgProc)
    await bot.connect()
    await bot.run()
    print("aa")

if __name__=='__main__':
    asyncio.run(main())