Edited: 1. agent.py - добавление базового функционала метода stream_responce. 2. Добавлены/отредактированы chat.js, style.css. 3. flask.py добавлен блок инициализации объектов OllamaProvider, Character, Agent

This commit is contained in:
2026-05-12 21:31:46 +03:00
parent 47a704e152
commit 3b4c8a1151
57 changed files with 6859 additions and 8 deletions
Binary file not shown.
+30
View File
@@ -32,6 +32,7 @@ There are some memories of your character:
User`s name: {self.user_name}
"""
'''def save_history(self):
os.makedirs("data/history", exist_ok=True)
path = f"data/history/{self.character.id}.json"
@@ -185,6 +186,35 @@ User`s name: {self.user_name}
return response
def stream_responce(self, user_input, temperature=None, max_tokens=None): #В ollamaapi нужно будет создать новый метод чисто на стриминг, пока что закомментил строки с цельным ответом
messages = self.build_messages(user_input)
if temperature is not None:
self.character.temperature = temperature
if max_tokens is not None:
self.character.max_tokens = max_tokens
self.character.save()
response = self.llm.generate_stream(
messages,
temperature=self.character.temperature,
max_tokens=self.character.max_tokens
)
# сохраняем историю
self.chat_history.append({
"role": "user",
"content": user_input
})
self.chat_history.append({
"role": "assistant",
"content": response
})
if len(self.chat_history) > 20:
self.summarize_history()
return response
"""
Пример использования agent`а (пример вызовов):
Binary file not shown.
+2 -1
View File
@@ -34,7 +34,8 @@ class Character():
self.first_message = first_message
self.temperature = temperature
self.max_tokens = max_tokens
# Methods:
#методы:
def to_dict(self):
return {
Binary file not shown.
+61 -4
View File
@@ -1,13 +1,70 @@
from flask import *
from flask import Flask, render_template, request, jsonify
from core.agent.agent import Agent
from core.llm.ollamaapi import OllamaProvider
from core.character.character import Character
ui = Flask(__name__)
#блок инициализации объектов
llm = OllamaProvider("gemma3:4b")
character = Character.load("test_char_001")
agent = Agent(character, llm, user_name="Alex")
agent.load_memory()
agent.ensure_first_message()
#база
@ui.route("/")
@ui.route("/index")
def index():
return render_template('index.html')
@ui.route("/chats")
def chats():
return "<h1>Страница чатов</h1>"
#история чата
@ui.route("/init", methods=["GET"])
def init():
return jsonify({
"messages": agent.chat_history,
"user_name": agent.user_name,
"character": {
"name": agent.character.name,
"avatar": agent.character.avatar_path
}
})
#запрос на генерацию
@ui.route("/chat", methods=["POST"])
def chat():
data = request.json
user_message = data["message"]
response = agent.respond(user_message)
agent.save_memory()
return jsonify({
"response": response
})
"""#список персонажей
@ui.route("/characters", methods=["GET"])
def get_characters():
return Agent.get_all_char_info()
#выбор персонажа (не робит)
@ui.route("/select_character", methods=["POST"])
def select_character():
pass
'''data = request.json
char_id = data["id"]
global agent
agent = Agent(characters[char_id], llm)
return jsonify({"status": "ok"})'''"""
+18 -2
View File
@@ -1,5 +1,5 @@
const sendBtn = document.getElementById("send-btn");
loadChat();
sendBtn.addEventListener("click", async () => {
const input = document.getElementById("user-input");
@@ -8,7 +8,11 @@ sendBtn.addEventListener("click", async () => {
const messages = document.getElementById("messages");
messages.innerHTML += `<p><b>You:</b> ${text}</p>`;
const source = new EventSource("/stream");
source.onmessage = function(event) {
messages.innerHTML += event.data;
};
const response = await fetch("/chat", {
method: "POST",
headers: {
@@ -24,4 +28,16 @@ sendBtn.addEventListener("click", async () => {
messages.innerHTML += `<p><b>AI:</b> ${data.response}</p>`;
input.value = "";
});
});
async function loadChat() {
const res = await fetch("/init");
const data = await res.json();
const messages = document.getElementById("messages");
data.messages.forEach(msg => {
messages.innerHTML += `<p><b>${msg.role}:</b> ${msg.content}</p>`;
});
}
Binary file not shown.
-1
View File
@@ -25,7 +25,6 @@ class OllamaProvider:
content = chunk['message']['content']
print(content, end='', flush=True)
response_text += content
return response_text
except Exception as e: