ken/agent
3
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

example stream 方式

Browse Source
This commit is contained in:
ken 2025-03-30 14:14:11 +08:00
parent e60fa21a0f
commit d86ca64fdf
10 changed files with 379 additions and 144 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/src/common.ts
View File

@ -1,8 +1,8 @@
import ws from 'ws';
process.env["NODE_TLS_REJECT_UNAUTHORIZED"]="0"
const ADDR = "ws://112.74.39.99:8080/ws"
// const ADDR = "ws://127.0.0.1:8080/ws"
// const ADDR = "ws://112.74.39.99:8080/ws"
const ADDR = "ws://127.0.0.1:8080/ws"
export function makeClient(
onMessage: ((data: ws.RawData) => void) | any = undefined,

16
examples/src/exec_chat.ts
View File

@ -1,9 +1,6 @@
import ws from 'ws';
import { makeClient } from "./common";
async function run() {
let client = await makeClient()
let req = {
"cmd" : "exec_chat",
"request_id" : 123, // 随机生成 （必填）
@ -13,6 +10,9 @@ async function run() {
"stream" : false // stream方式返回，目前未支持 （可选）
}
async function run() {
let client = await makeClient()
req.stream = false
client.send(JSON.stringify(req))
}
@ -24,15 +24,7 @@ async function run_stream() {
// console.log(data.toString())
})
let req = {
"cmd" : "exec_chat",
"request_id" : 123, // 随机生成 （必填）
"language" : "en", // 语言, en-英文, cn-中文，默认为 en （可选）
"msg" : "What would be a good company name a company that makes colorful socks?",
"model" : "local qwen2.5-coder:7b", //（必填）
"stream" : true // stream方式返回，目前未支持 （可选）
}
req.stream = true
client.send(JSON.stringify(req))
}

24
examples/src/exec_docstring.ts
View File

@ -1,8 +1,6 @@
import ws from 'ws';
import { makeClient } from "./common";
async function run() {
let client = await makeClient()
let selected_text = {
"filepath": "example\\\\game.py", // 文件路径
"range": {
@ -15,11 +13,27 @@ async function run() {
let req = {
"cmd" : "exec_docstring",
"request_id": 123, // 随机生成 （必填）
"model" : "local deepseek-r1:32b", //（必填）
"model" : "local qwen2.5-coder:7b", //（必填）
"stream" : false,
"selected_text": selected_text,
}
async function run() {
let client = await makeClient()
req.stream = false
client.send(JSON.stringify(req))
}
run()
async function run_stream() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
process.stdout.write(d.msg)
// console.log(data.toString())
})
req.stream = true
client.send(JSON.stringify(req))
}
run_stream()

22
examples/src/exec_explain.ts
View File

@ -1,8 +1,6 @@
import ws from 'ws';
import { makeClient } from "./common";
async function run() {
let client = await makeClient()
let selected_text = {
"filepath": "example\\\\game.py", // 文件路径
"range": {
@ -29,7 +27,23 @@ async function run() {
"selected_text": selected_text,
"visible_text" : visible_text
}
async function run() {
let client = await makeClient()
req.stream = false
client.send(JSON.stringify(req))
}
run()
async function run_stream() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
process.stdout.write(d.msg)
// console.log(data.toString())
})
req.stream = true
client.send(JSON.stringify(req))
}
run_stream()

25
examples/src/exec_fix.ts
View File

@ -1,14 +1,6 @@
import ws from 'ws';
import { makeClient } from "./common";
async function run() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
console.log("------------------------------- msg start -------------------------------")
console.log(d.msg)
console.log("------------------------------- msg end -------------------------------")
})
let selected_text = {
"filepath": "example\\\\game.py", // 文件路径
"range": {
@ -25,7 +17,22 @@ async function run() {
"stream" : false,
"selected_text": selected_text,
}
async function run() {
let client = await makeClient()
req.stream = false
client.send(JSON.stringify(req))
}
run()
async function run_stream() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
process.stdout.write(d.msg)
// console.log(data.toString())
})
req.stream = true
client.send(JSON.stringify(req))
}
run_stream()

26
examples/src/exec_optimize.ts
View File

@ -1,14 +1,6 @@
import ws from 'ws';
import { makeClient } from "./common";
async function run() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
console.log("------------------------------- msg start -------------------------------")
console.log(d.msg)
console.log("------------------------------- msg end -------------------------------")
})
let selected_text = {
"filepath": "example\\\\game.py", // 文件路径
"range": {
@ -25,7 +17,23 @@ async function run() {
"stream" : false,
"selected_text": selected_text,
}
async function run() {
let client = await makeClient()
req.stream = false
client.send(JSON.stringify(req))
}
run()
async function run_stream() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
process.stdout.write(d.msg)
// console.log(data.toString())
})
req.stream = true
client.send(JSON.stringify(req))
}
run_stream()

26
examples/src/exec_unitest.ts
View File

@ -1,14 +1,6 @@
import ws from 'ws';
import { makeClient } from "./common";
async function run() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
console.log("------------------------------- msg start -------------------------------")
console.log(d.msg)
console.log("------------------------------- msg end -------------------------------")
})
let selected_text = {
"filepath": "example\\\\game.py", // 文件路径
"range": {
@ -25,7 +17,23 @@ async function run() {
"stream" : false,
"selected_text": selected_text,
}
async function run() {
let client = await makeClient()
req.stream = false
client.send(JSON.stringify(req))
}
run()
async function run_stream() {
let client = await makeClient((data: ws.RawData)=>{
let d = JSON.parse(data.toString())
process.stdout.write(d.msg)
// console.log(data.toString())
})
req.stream = true
client.send(JSON.stringify(req))
}
run_stream()

188
examples/src/game.py Normal file
View File

@ -0,0 +1,188 @@
# Program in Python to create a Snake Game
from tkinter import *
import random
# Initialising Dimensions of Game
WIDTH = 500
HEIGHT = 500
SPEED = 200
SPACE_SIZE = 20
BODY_SIZE = 2
SNAKE = "#0000FF"
FOOD = "#FFFF00"
BACKGROUND = "#000000"
# Class to design the snake
class Snake:
def __init__(self):
self.body_size = BODY_SIZE
self.coordinates = []
self.squares = []
for i in range(0, BODY_SIZE):
self.coordinates.append([0, 0])
for x, y in self.coordinates:
square = canvas.create_rectangle(
x, y, x + SPACE_SIZE, y + SPACE_SIZE,
fill=SNAKE, tag="snake")
self.squares.append(square)
# Class to design the food
class Food:
def __init__(self):
x = random.randint(0,
(WIDTH / SPACE_SIZE)-1) * SPACE_SIZE
y = random.randint(0,
(HEIGHT / SPACE_SIZE) - 1) * SPACE_SIZE
self.coordinates = [x, y]
canvas.create_oval(x, y, x + SPACE_SIZE, y +
SPACE_SIZE, fill=FOOD, tag="food")
# Function to check the next move of snake
def next_turn(snake, food):
x, y = snake.coordinates[0]
if direction == "up":
y -= SPACE_SIZE
elif direction == "down":
y += SPACE_SIZE
elif direction == "left":
x -= SPACE_SIZE
elif direction == "right":
x += SPACE_SIZE
snake.coordinates.insert(0, (x, y))
square = canvas.create_rectangle(
x, y, x + SPACE_SIZE,
y + SPACE_SIZE, fill=SNAKE)
snake.squares.insert(0, square)
if x == food.coordinates[0] and y == food.coordinates[1]:
global score
score += 1
label.config(text="Points:{}".format(score))
canvas.delete("food")
food = Food()
else:
del snake.coordinates[-1]
canvas.delete(snake.squares[-1])
del snake.squares[-1]
if check_collisions(snake):
game_over()
else:
window.after(SPEED, next_turn, snake, food)
# Function to control direction of snake
def change_direction(new_direction):
global direction
if new_direction == 'left':
if direction != 'right':
direction = new_direction
elif new_direction == 'right':
if direction != 'left':
direction = new_direction
elif new_direction == 'up':
if direction != 'down':
direction = new_direction
elif new_direction == 'down':
if direction != 'up':
direction = new_direction
# function to check snake's collision and position
def check_collisions(snake):
x, y = snake.coordinates[0]
if x < 0 or x >= WIDTH:
return True
elif y < 0 or y >= HEIGHT:
return True
for body_part in snake.coordinates[1:]:
if x == body_part[0] and y == body_part[1]:
return True
return False
# Function to control everything
def game_over():
canvas.delete(ALL)
canvas.create_text(canvas.winfo_width()/2,
canvas.winfo_height()/2,
font=('consolas', 70),
text="GAME OVER", fill="red",
tag="gameover")
# Giving title to the gaming window
window = Tk()
window.title("Snake Game")
score = 0
direction = 'down'
# Display of Points Scored in Game
label = Label(window, text="Points:{}".format(score),
font=('consolas', 20))
label.pack()
canvas = Canvas(window, bg=BACKGROUND,
height=HEIGHT, width=WIDTH)
canvas.pack()
window.update()
window_width = window.winfo_width()
window_height = window.winfo_height()
screen_width = window.winfo_screenwidth()
screen_height = window.winfo_screenheight()
x = int((screen_width/2) - (window_width/2))
y = int((screen_height/2) - (window_height/2))
window.geometry(f"{window_width}x{window_height}+{x}+{y}")
window.bind('<Left>',
lambda event: change_direction('left'))
window.bind('<Right>',
lambda event: change_direction('right'))
window.bind('<Up>',
lambda event: change_direction('up'))
window.bind('<Down>',
lambda event: change_direction('down'))
snake = Snake()
food = Food()
next_turn(snake, food)
window.mainloop()
# This code is contributed by genius_general

12
src/llm/llm.go
View File

@ -30,10 +30,11 @@ func InitLocalModel(urls []string) error {
LLMNames = make([]string, 0, len(model2url))
LLMGroups = make(map[string]*LLMGroup, len(model2url))
var models []string
for model, urls := range model2url {
model = "local " + model
localModel := "local " + model
g := &LLMGroup{
name: model,
name: localModel,
llm: make(chan llms.Model, len(urls)),
local: true,
}
@ -46,10 +47,11 @@ func InitLocalModel(urls []string) error {
}
LLMGroups[model] = g
LLMNames = append(LLMNames, model)
LLMGroups[localModel] = g
LLMNames = append(LLMNames, localModel)
models = append(models, model)
}
log.Info("load models", "models", strings.Join(LLMNames, ","))
log.Info("load models", "models", strings.Join(models, ","))
return nil
}

2
src/workflow/exec.go
View File

@ -3,6 +3,7 @@ package workflow
import (
"agent/src/message"
"agent/src/prompt"
"agent/src/utils/log"
)
@ -24,6 +25,7 @@ func (task *Task) explain() error {
}
func (task *Task) docstring() error {
log.Info("call docstring")
req := task.request.(*message.RequestExecDocstring)
contents := prompt.MakeExecDocstring(req.Language, req.SelectedText.FilePath, req.SelectedText.Text)