admin/Project02
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新通信协议

Browse Source
This commit is contained in:
997146918 2025-07-07 09:33:56 +08:00
parent 590970b0b3
commit 5737e415c9
8 changed files with 180 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
AIGC/DatabaseHandle.py
View File

@ -52,7 +52,12 @@ class DatabaseHandle:
name, age, personality, profession, characterBackground,
chat_style
) VALUES (?, ?, ?, ?, ?, ?)
ON CONFLICT(name) DO NOTHING
ON CONFLICT(name) DO UPDATE SET
age = excluded.age,
personality = excluded.personality,
profession = excluded.profession,
characterBackground = excluded.characterBackground,
chat_style = excluded.chat_style
''', (
data["name"], data["age"], data["personality"],
data["profession"], data["characterBackground"],

BIN
AIGC/data.db Normal file
View File

Binary file not shown.

119
AIGC/main.py
View File

@ -42,14 +42,10 @@ async def heartbeat(websocket: WebSocket):
# break # 连接已关闭时退出循环
#statuscode -1 服务器运行错误 0 心跳标志 1 正常 2 输出异常
async def senddata(websocket: WebSocket, statusCode: int, messages: List[str]):
async def senddata(websocket: WebSocket, protocol: dict):
# 将AI响应发送回UE5
if websocket.client_state == WebSocketState.CONNECTED:
data = {
"statuscode": statusCode,
"messages": messages
}
json_string = json.dumps(data, ensure_ascii=False)
json_string = json.dumps(protocol, ensure_ascii=False)
await websocket.send_text(json_string)
# WebSocket路由处理
@ -65,15 +61,17 @@ async def websocket_endpoint(websocket: WebSocket, client_id: str):
# 接收UE5发来的消息
data = await websocket.receive_text()
logger.log(logging.INFO, f"收到UE5消息 [{client_id}]: {data}")
success, prompt = process_prompt(data)
global lastPrompt
lastPrompt = prompt
# 调用AI生成响应
if(success):
asyncio.create_task(generateAIChat(prompt, websocket))
await senddata(websocket, 0, [])
else:
await senddata(websocket, -1, [])
await process_protocol_json(data, websocket)
# success, prompt = process_prompt(data)
# global lastPrompt
# lastPrompt = prompt
# # 调用AI生成响应
# if(success):
# asyncio.create_task(generateAIChat(prompt, websocket))
# await senddata(websocket, 0, [])
# else:
# await senddata(websocket, -1, [])
@ -85,6 +83,29 @@ async def websocket_endpoint(websocket: WebSocket, client_id: str):
#manager.disconnect(client_id)
logger.log(logging.ERROR, f"WebSocket异常 [{client_id}]: {str(e)}")
async def handle_characterlist(client: WebSocket):
### 获得数据库中的角色信息###
characters = database.get_character_byname("")
protocol = {}
protocol["cmd"] = "CharacterList"
protocol["status"] = 1
protocol["message"] = "success"
protocol["data"] = json.dumps(characters)
await senddata(client, protocol)
async def process_protocol_json(json_str: str, client: WebSocket):
### 处理协议JSON ###
try:
protocol = json.loads(json_str)
cmd = protocol.get("cmd")
data = protocol.get("data")
if cmd == "CharacterList":
await handle_characterlist(client)
except json.JSONDecodeError as e:
print(f"JSON解析错误: {e}")
def process_prompt(promptFromUE: str) -> Tuple[bool, str]:
try:
data = json.loads(promptFromUE)
@ -253,42 +274,42 @@ if __name__ == "__main__":
logger.log(logging.ERROR, f"角色 张三已经添加到数据库")
# Test AI
aicore.getPromptToken("测试功能")
asyncio.run(
generateAIChat(promptStr = f"""
#你是一个游戏NPC对话生成器。请严格按以下要求生成两个角色的日常对话
#对话的世界观背景是2025年的都市背景
1. 生成【2轮完整对话】，每轮包含双方各一次发言（共4句）
2.角色设定
# asyncio.run(
# generateAIChat(promptStr = f"""
# #你是一个游戏NPC对话生成器。请严格按以下要求生成两个角色的日常对话
# #对话的世界观背景是2025年的都市背景
# 1. 生成【2轮完整对话】，每轮包含双方各一次发言（共4句）
# 2.角色设定
"张三": {{
"姓名": "张三",
"年龄": 35,
"性格": "成熟稳重/惜字如金",
"职业": "阿里巴巴算法工程师",
"背景": "浙大计算机系毕业，专注AI优化项目",
"对话场景": "你正在和用户聊天，用户是你的同事",
"语言风格": "请在对话中表现出专业、冷静、惜字如金。用口语化的方式简短回答"
}},
"李明": {{
"姓名": "李明",
"年龄": 30,
"职业": "产品经理",
"性格": "活泼健谈"
"背景": "公司资深产品经理",
"对话场景": "你正在和用户聊天，用户是你的同事",
"语言风格": "热情"
}}
# "张三": {{
# "姓名": "张三",
# "年龄": 35,
# "性格": "成熟稳重/惜字如金",
# "职业": "阿里巴巴算法工程师",
# "背景": "浙大计算机系毕业，专注AI优化项目",
# "对话场景": "你正在和用户聊天，用户是你的同事",
# "语言风格": "请在对话中表现出专业、冷静、惜字如金。用口语化的方式简短回答"
# }},
# "李明": {{
# "姓名": "李明",
# "年龄": 30,
# "职业": "产品经理",
# "性格": "活泼健谈"
# "背景": "公司资深产品经理",
# "对话场景": "你正在和用户聊天，用户是你的同事",
# "语言风格": "热情"
# }}
3.输出格式：
<format>
张三：[第一轮发言]
李明：[第一轮回应]
张三：[第二轮发言]
李明：[第二轮回应]
</format>
"""
)
)
# 3.输出格式：
# <format>
# 张三：[第一轮发言]
# 李明：[第一轮回应]
# 张三：[第二轮发言]
# 李明：[第二轮回应]
# </format>
# """
# )
# )
try:
# 主线程永久挂起（监听退出信号）

5
TestForAIGC/Plugins/AIGC/Source/AIGC/Definations.cpp Normal file
View File

@ -0,0 +1,5 @@
#include "Definations.h"
FString FNetCommand::CharacterList = TEXT("CharacterList");
FString FNetCommand::AddCharacter = TEXT("AddCharacter");
FString FNetCommand::AiChatGenerate = TEXT("AiChatGenerate");

4
TestForAIGC/Plugins/AIGC/Source/AIGC/Private/AIGCSetting.cpp Normal file
View File

@ -0,0 +1,4 @@
// Fill out your copyright notice in the Description page of Project Settings.
#include "AIGCSetting.h"

53
TestForAIGC/Plugins/AIGC/Source/AIGC/Private/Widget/CharacterWindow.cpp Normal file
View File

@ -0,0 +1,53 @@
// Fill out your copyright notice in the Description page of Project Settings.
#include "Widget/CharacterWindow.h"
#include "AIGC.h"
#include "WebSocketManager.h"
void SCharacterWindow::Construct(const FArguments& InArgs)
{
// 注册绘制后回调
RegisterActiveTimer(0.f,
FWidgetActiveTimerDelegate::CreateSP(
this,
&SCharacterWindow::OnPostPaint
)
);
}
EActiveTimerReturnType SCharacterWindow::OnPostPaint(double X, float Arg)
{
//init websocket
FAIGCModule* ModulePtr = FModuleManager::GetModulePtr<FAIGCModule>("AIGC");
if (ModulePtr)
{
UWebSocketManager* WebSocketManager = ModulePtr->GetWebSocketManager();
if (!WebSocketManager)
{
ModulePtr->InitWebSocketManager();
WebSocketManager = ModulePtr->GetWebSocketManager();
}
else
{
WebSocketManager->SendData(FNetCommand::CharacterList, TEXT(""));
}
WebSocketManager->OnDataReceiveDelaget.AddRaw(this, &SCharacterWindow::HandleReceiveData);
WebSocketManager->OnConnectDelegate.AddLambda([this, WebSocketManager](bool bSuccess)
{
WebSocketManager->SendData(FNetCommand::CharacterList, TEXT(""));
});
}
return EActiveTimerReturnType::Stop;
}
void SCharacterWindow::HandleReceiveData(FNetProtocol protocol)
{
if (protocol.cmd == FNetCommand::CharacterList)
{
//刷新角色列表
}
}

19
TestForAIGC/Plugins/AIGC/Source/AIGC/Public/AIGCSetting.h Normal file
View File

@ -0,0 +1,19 @@
// Fill out your copyright notice in the Description page of Project Settings.
#pragma once
#include "CoreMinimal.h"
#include "UObject/Object.h"
#include "AIGCSetting.generated.h"
/**
*
*/
UCLASS(Config=AIGCSetting, DefaultConfig)
class AIGC_API UAIGCSetting : public UObject
{
GENERATED_BODY()
public:
UPROPERTY(Config, EditAnywhere)
FString ServerIP = TEXT("127.0.0.1");
};

23
TestForAIGC/Plugins/AIGC/Source/AIGC/Public/Widget/CharacterWindow.h Normal file
View File

@ -0,0 +1,23 @@
// Fill out your copyright notice in the Description page of Project Settings.
#pragma once
#include "CoreMinimal.h"
#include "Definations.h"
/**
*
*/
class AIGC_API SCharacterWindow: public SCompoundWidget
{
public:
SLATE_BEGIN_ARGS(SCharacterWindow)
: _CharactersJsons()
{}
SLATE_ARGUMENT(FString, CharactersJsons)
SLATE_END_ARGS()
void Construct(const FArguments& InArgs);
EActiveTimerReturnType OnPostPaint(double X, float Arg);
void HandleReceiveData(FNetProtocol protocol);
};