Add pre commit (#26)

This commit is contained in:
Lamont Huffman
2025-11-05 11:44:19 +08:00
committed by GitHub
parent 2164371e64
commit 30d86efbb3
33 changed files with 565 additions and 496 deletions

View File

@@ -3,8 +3,8 @@ import asyncio
import os
from agentscope.agent import ReActAgent
from agentscope_runtime.engine import LocalDeployManager, Runner
from agentscope.model import DashScopeChatModel
from agentscope_runtime.engine import LocalDeployManager, Runner
from agentscope_runtime.engine.agents.agentscope_agent import AgentScopeAgent
from agentscope_runtime.engine.services.context_manager import ContextManager
@@ -23,12 +23,13 @@ async def _local_deploy():
model = DashScopeChatModel(
model_name="qwen-turbo",
api_key=os.getenv("DASHSCOPE_API_KEY"),
)
agent = AgentScopeAgent(
name="Friday",
model=model,
agent_config={"sys_prompt": "A simple LLM agent to generate a short response"},
agent_config={
"sys_prompt": "A simple LLM agent to generate a short response",
},
agent_builder=ReActAgent,
)

View File

@@ -3,14 +3,13 @@ import json
import logging
import os
from datetime import datetime
from typing import Tuple, Optional, Union, Dict, Any, Generator
import requests
from dotenv import load_dotenv
from flask import Flask, jsonify, request
from flask import Flask, request, jsonify
from flask_cors import CORS
from flask_sqlalchemy import SQLAlchemy
from werkzeug.security import check_password_hash, generate_password_hash
from werkzeug.security import generate_password_hash, check_password_hash
logging.basicConfig(
level=logging.INFO,
@@ -48,10 +47,10 @@ class User(db.Model):
cascade="all, delete-orphan",
)
def set_password(self, password: str) -> None:
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def check_password(self, password: str) -> bool:
def check_password(self, password):
return check_password_hash(self.password_hash, password)
@@ -90,7 +89,7 @@ class Message(db.Model):
# Create database tables
def create_tables() -> None:
def create_tables():
db.create_all()
# Create sample users (if none exist)
@@ -105,9 +104,7 @@ def create_tables() -> None:
# functions
def parse_sse_line(
line: bytes,
) -> Tuple[Optional[str], Optional[Union[str, int]]]:
def parse_sse_line(line):
line = line.decode("utf-8").strip()
if line.startswith("data: "):
return "data", line[6:]
@@ -120,10 +117,7 @@ def parse_sse_line(
return None, None
def sse_client(
url: str,
data: Optional[Dict[str, Any]] = None,
) -> Generator[str, None, None]:
def sse_client(url, data=None):
headers = {
"Accept": "text/event-stream",
"Cache-Control": "no-cache",
@@ -158,17 +152,13 @@ def sse_client(
pass
def call_runner(
query: str,
query_user_id: str,
query_session_id: str,
) -> Generator[str, None, None]:
def call_runner(query, query_user_id, query_session_id):
server_port = int(os.environ.get("SERVER_PORT", "8090"))
server_endpoint = os.environ.get("SERVER_ENDPOINT", "agent")
server_host = os.environ.get("SERVER_HOST", "localhost")
url = f"http://{server_host}:{server_port}/{server_endpoint}"
data_arg: Dict[str, Any] = {
data_arg = {
"input": [
{
"role": "user",