Initial Kaya board-mention agent daemon.
All checks were successful
ci/woodpecker/manual/woodpecker Pipeline was successful
All checks were successful
ci/woodpecker/manual/woodpecker Pipeline was successful
Thin WS client to l-ai-tasks that drafts short core RunAgent replies on @kaya mentions. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
alex
commit
adb6c5ab8a
17
.env.example
Normal file
17
.env.example
Normal file
@ -0,0 +1,17 @@
|
||||
# Tasks board (WebSocket + REST)
|
||||
TASKS_WS_URL=ws://localhost:8080/api/ws
|
||||
TASKS_API_URL=http://localhost:8080
|
||||
TASKS_AGENT_TOKEN=
|
||||
|
||||
# l-ai-core AIService.RunAgent
|
||||
CORE_ENDPOINT=localhost:50051
|
||||
CORE_TOKEN=
|
||||
CORE_INSECURE=true
|
||||
|
||||
MODEL=deepseek-v4-flash
|
||||
AGENT_PROFILE_PATH=profiles/kaya.yaml
|
||||
|
||||
HEALTH_LISTEN_ADDR=:8087
|
||||
METRICS_LISTEN_ADDR=:9097
|
||||
SHUTDOWN_TIMEOUT=15s
|
||||
RESPONSE_TIMEOUT=120s
|
||||
8
.gitignore
vendored
Normal file
8
.gitignore
vendored
Normal file
@ -0,0 +1,8 @@
|
||||
*.exe
|
||||
*.test
|
||||
*.out
|
||||
.env
|
||||
.idea/
|
||||
.vscode/
|
||||
bin/
|
||||
dist/
|
||||
3
.gitmodules
vendored
Normal file
3
.gitmodules
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
[submodule "proto/l-ai-core-proto"]
|
||||
path = proto/l-ai-core-proto
|
||||
url = https://gitea.fe.kaien.ru/kaien/l-ai-core-proto.git
|
||||
73
.woodpecker.yml
Normal file
73
.woodpecker.yml
Normal file
@ -0,0 +1,73 @@
|
||||
clone:
|
||||
git:
|
||||
image: woodpeckerci/plugin-git
|
||||
settings:
|
||||
submodule_recursive: true
|
||||
|
||||
steps:
|
||||
- name: test
|
||||
image: golang:1.25-alpine
|
||||
commands:
|
||||
- apk add --no-cache git build-base
|
||||
- go build ./...
|
||||
- go vet ./...
|
||||
- go test ./...
|
||||
when:
|
||||
- event: [push, pull_request, manual]
|
||||
|
||||
- name: build-and-push
|
||||
image: gcr.io/kaniko-project/executor:v1.23.2-debug
|
||||
environment:
|
||||
ZOT_USERNAME:
|
||||
from_secret: ZOT_USERNAME
|
||||
ZOT_PASSWORD:
|
||||
from_secret: ZOT_PASSWORD
|
||||
commands:
|
||||
- mkdir -p /kaniko/.docker
|
||||
- >-
|
||||
printf '{"auths":{"registry.be.kaien.ru":{"auth":"%s"}}}'
|
||||
"$(printf '%s:%s' "$ZOT_USERNAME" "$ZOT_PASSWORD" | base64 | tr -d '\n')"
|
||||
> /kaniko/.docker/config.json
|
||||
- >-
|
||||
/kaniko/executor
|
||||
--context "$CI_WORKSPACE"
|
||||
--dockerfile "$CI_WORKSPACE/Dockerfile"
|
||||
--destination "registry.be.kaien.ru/l_ai_agent_kaya/agent_kaya:$CI_COMMIT_SHA"
|
||||
--destination "registry.be.kaien.ru/l_ai_agent_kaya/agent_kaya:latest"
|
||||
depends_on:
|
||||
- test
|
||||
when:
|
||||
- branch: master
|
||||
event: [push, manual]
|
||||
|
||||
- name: deploy
|
||||
image: alpine:latest
|
||||
environment:
|
||||
SSH_PRIVATE_KEY:
|
||||
from_secret: SSH_PRIVATE_KEY
|
||||
DEPLOY_HOST:
|
||||
from_secret: SWARM_DEPLOY_HOST
|
||||
DEPLOY_USER:
|
||||
from_secret: SWARM_DEPLOY_USER
|
||||
commands:
|
||||
- apk add --no-cache openssh-client
|
||||
- eval $(ssh-agent -s)
|
||||
- mkdir -p ~/.ssh && chmod 700 ~/.ssh
|
||||
- echo "$SSH_PRIVATE_KEY" > ~/.ssh/id_rsa
|
||||
- chmod 600 ~/.ssh/id_rsa
|
||||
- ssh-add ~/.ssh/id_rsa
|
||||
- ssh-keyscan -H "$DEPLOY_HOST" >> ~/.ssh/known_hosts
|
||||
- chmod 644 ~/.ssh/known_hosts
|
||||
- >-
|
||||
ssh -n $DEPLOY_USER@$DEPLOY_HOST
|
||||
"docker service update
|
||||
--image registry.be.kaien.ru/l_ai_agent_kaya/agent_kaya:$CI_COMMIT_SHA
|
||||
--with-registry-auth
|
||||
--update-order start-first
|
||||
--detach=false
|
||||
l-ai_agent-kaya"
|
||||
depends_on:
|
||||
- build-and-push
|
||||
when:
|
||||
- branch: master
|
||||
event: [push, manual]
|
||||
17
Dockerfile
Normal file
17
Dockerfile
Normal file
@ -0,0 +1,17 @@
|
||||
FROM golang:1.25-alpine AS build
|
||||
WORKDIR /app
|
||||
|
||||
COPY go.mod go.sum ./
|
||||
RUN apk add --no-cache git
|
||||
RUN go mod download
|
||||
|
||||
COPY . .
|
||||
RUN CGO_ENABLED=0 go build -o /kaya ./cmd/kaya
|
||||
|
||||
FROM alpine:3.20
|
||||
RUN apk add --no-cache ca-certificates
|
||||
WORKDIR /app
|
||||
COPY --from=build /kaya /app/kaya
|
||||
COPY profiles /app/profiles
|
||||
EXPOSE 8087 9097
|
||||
ENTRYPOINT ["/app/kaya"]
|
||||
45
README.md
Normal file
45
README.md
Normal file
@ -0,0 +1,45 @@
|
||||
# l-ai-agent-kaya
|
||||
|
||||
Thin Tasks board agent: replies to `@kaya` mentions in task comments.
|
||||
|
||||
## Behavior
|
||||
|
||||
Long-lived daemon:
|
||||
|
||||
1. Connects to `l-ai-tasks` WebSocket (`TASKS_WS_URL`) with reconnect.
|
||||
2. On `comment.added` where mentions include agent `kaya` and author ≠ `kaya`:
|
||||
- dedupes by comment id
|
||||
- GET task + comments via REST
|
||||
- calls `l-ai-core` `AIService.RunAgent` (tools disabled) with Kaya persona
|
||||
- POST comment with `Idempotency-Key: mention-reply:<comment_id>`
|
||||
|
||||
No Telegram / Live Chat integration.
|
||||
|
||||
## Identity
|
||||
|
||||
- `agent_id`: `kaya`
|
||||
- Display: Kaya / Кая
|
||||
- Profile: `profiles/kaya.yaml`
|
||||
|
||||
## Env
|
||||
|
||||
| Variable | Required | Default |
|
||||
|---|---|---|
|
||||
| `TASKS_WS_URL` | yes | — |
|
||||
| `TASKS_API_URL` | yes | — |
|
||||
| `TASKS_AGENT_TOKEN` | yes | — |
|
||||
| `CORE_ENDPOINT` | no | `core:50051` |
|
||||
| `CORE_TOKEN` | yes | — |
|
||||
| `CORE_INSECURE` | no | `false` |
|
||||
| `MODEL` | no | `deepseek-v4-flash` |
|
||||
| `AGENT_PROFILE_PATH` | no | `profiles/kaya.yaml` |
|
||||
| `HEALTH_LISTEN_ADDR` | no | `:8087` |
|
||||
| `METRICS_LISTEN_ADDR` | no | `:9097` |
|
||||
|
||||
## Build
|
||||
|
||||
```bash
|
||||
go build -o kaya ./cmd/kaya
|
||||
```
|
||||
|
||||
Image binary: `/app/kaya`. Health `:8087/healthz`, metrics `:9097/metrics`.
|
||||
86
cmd/kaya/main.go
Normal file
86
cmd/kaya/main.go
Normal file
@ -0,0 +1,86 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||
"l-ai-agent-kaya/internal/config"
|
||||
"l-ai-agent-kaya/internal/inference"
|
||||
"l-ai-agent-kaya/internal/profile"
|
||||
"l-ai-agent-kaya/internal/tasksboard"
|
||||
)
|
||||
|
||||
func main() {
|
||||
if err := run(); err != nil {
|
||||
slog.Error("kaya stopped", "error", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func run() error {
|
||||
cfg, err := config.Load()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
prof, err := profile.Load(cfg.ProfilePath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||||
defer stop()
|
||||
|
||||
core, err := inference.DialCore(cfg.CoreEndpoint, cfg.CoreToken, cfg.CoreInsecure)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer core.Close()
|
||||
|
||||
go serveHTTP(cfg.MetricsListenAddr, promhttp.Handler())
|
||||
go serveHTTP(cfg.HealthListenAddr, healthHandler())
|
||||
|
||||
loop := &tasksboard.Loop{
|
||||
WSURL: cfg.TasksWSURL,
|
||||
Token: cfg.TasksAgentToken,
|
||||
REST: &tasksboard.Client{BaseURL: cfg.TasksAPIURL, Token: cfg.TasksAgentToken},
|
||||
Core: core,
|
||||
Profile: prof,
|
||||
Model: cfg.Model,
|
||||
ResponseTimeout: cfg.ResponseTimeout,
|
||||
Log: slog.Default(),
|
||||
}
|
||||
go loop.Run(ctx)
|
||||
|
||||
<-ctx.Done()
|
||||
shutdown, cancel := context.WithTimeout(context.Background(), cfg.ShutdownTimeout)
|
||||
defer cancel()
|
||||
<-shutdown.Done()
|
||||
if errors.Is(ctx.Err(), context.Canceled) {
|
||||
return nil
|
||||
}
|
||||
return ctx.Err()
|
||||
}
|
||||
|
||||
func serveHTTP(addr string, h http.Handler) {
|
||||
if addr == "" {
|
||||
return
|
||||
}
|
||||
if err := http.ListenAndServe(addr, h); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||
slog.Error("http server stopped", "addr", addr, "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
func healthHandler() http.Handler {
|
||||
mux := http.NewServeMux()
|
||||
mux.HandleFunc("/healthz", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
_, _ = w.Write([]byte("ok\n"))
|
||||
})
|
||||
return mux
|
||||
}
|
||||
26
go.mod
Normal file
26
go.mod
Normal file
@ -0,0 +1,26 @@
|
||||
module l-ai-agent-kaya
|
||||
|
||||
go 1.25.0
|
||||
|
||||
require (
|
||||
github.com/gorilla/websocket v1.5.3
|
||||
github.com/prometheus/client_golang v1.23.2
|
||||
google.golang.org/grpc v1.70.0
|
||||
google.golang.org/protobuf v1.36.8
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/beorn7/perks v1.0.1 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||
github.com/kr/text v0.2.0 // indirect
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
|
||||
github.com/prometheus/client_model v0.6.2 // indirect
|
||||
github.com/prometheus/common v0.66.1 // indirect
|
||||
github.com/prometheus/procfs v0.16.1 // indirect
|
||||
go.yaml.in/yaml/v2 v2.4.2 // indirect
|
||||
golang.org/x/net v0.43.0 // indirect
|
||||
golang.org/x/sys v0.35.0 // indirect
|
||||
golang.org/x/text v0.28.0 // indirect
|
||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20241202173237-19429a94021a // indirect
|
||||
)
|
||||
74
go.sum
Normal file
74
go.sum
Normal file
@ -0,0 +1,74 @@
|
||||
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=
|
||||
github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
|
||||
github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
|
||||
github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
|
||||
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
||||
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
||||
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
||||
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||
github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=
|
||||
github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=
|
||||
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
|
||||
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
|
||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||
github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc=
|
||||
github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw=
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 h1:C3w9PqII01/Oq1c1nUAm88MOHcQC9l5mIlSMApZMrHA=
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/prometheus/client_golang v1.23.2 h1:Je96obch5RDVy3FDMndoUsjAhG5Edi49h0RJWRi/o0o=
|
||||
github.com/prometheus/client_golang v1.23.2/go.mod h1:Tb1a6LWHB3/SPIzCoaDXI4I8UHKeFTEQ1YCr+0Gyqmg=
|
||||
github.com/prometheus/client_model v0.6.2 h1:oBsgwpGs7iVziMvrGhE53c/GrLUsZdHnqNwqPLxwZyk=
|
||||
github.com/prometheus/client_model v0.6.2/go.mod h1:y3m2F6Gdpfy6Ut/GBsUqTWZqCUvMVzSfMLjcu6wAwpE=
|
||||
github.com/prometheus/common v0.66.1 h1:h5E0h5/Y8niHc5DlaLlWLArTQI7tMrsfQjHV+d9ZoGs=
|
||||
github.com/prometheus/common v0.66.1/go.mod h1:gcaUsgf3KfRSwHY4dIMXLPV0K/Wg1oZ8+SbZk/HH/dA=
|
||||
github.com/prometheus/procfs v0.16.1 h1:hZ15bTNuirocR6u0JZ6BAHHmwS1p8B4P6MRqxtzMyRg=
|
||||
github.com/prometheus/procfs v0.16.1/go.mod h1:teAbpZRB1iIAJYREa1LsoWUXykVXA1KlTmWl8x/U+Is=
|
||||
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||
github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=
|
||||
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
|
||||
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
|
||||
go.opentelemetry.io/otel v1.32.0 h1:WnBN+Xjcteh0zdk01SVqV55d/m62NJLJdIyb4y/WO5U=
|
||||
go.opentelemetry.io/otel v1.32.0/go.mod h1:00DCVSB0RQcnzlwyTfqtxSm+DRr9hpYrHjNGiBHVQIg=
|
||||
go.opentelemetry.io/otel/metric v1.32.0 h1:xV2umtmNcThh2/a/aCP+h64Xx5wsj8qqnkYZktzNa0M=
|
||||
go.opentelemetry.io/otel/metric v1.32.0/go.mod h1:jH7CIbbK6SH2V2wE16W05BHCtIDzauciCRLoc/SyMv8=
|
||||
go.opentelemetry.io/otel/sdk v1.32.0 h1:RNxepc9vK59A8XsgZQouW8ue8Gkb4jpWtJm9ge5lEG4=
|
||||
go.opentelemetry.io/otel/sdk v1.32.0/go.mod h1:LqgegDBjKMmb2GC6/PrTnteJG39I8/vJCAP9LlJXEjU=
|
||||
go.opentelemetry.io/otel/sdk/metric v1.32.0 h1:rZvFnvmvawYb0alrYkjraqJq0Z4ZUJAiyYCU9snn1CU=
|
||||
go.opentelemetry.io/otel/sdk/metric v1.32.0/go.mod h1:PWeZlq0zt9YkYAp3gjKZ0eicRYvOh1Gd+X99x6GHpCQ=
|
||||
go.opentelemetry.io/otel/trace v1.32.0 h1:WIC9mYrXf8TmY/EXuULKc8hR17vE+Hjv2cssQDe03fM=
|
||||
go.opentelemetry.io/otel/trace v1.32.0/go.mod h1:+i4rkvCraA+tG6AzwloGaCtkx53Fa+L+V8e9a7YvhT8=
|
||||
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
|
||||
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
|
||||
go.yaml.in/yaml/v2 v2.4.2 h1:DzmwEr2rDGHl7lsFgAHxmNz/1NlQ7xLIrlN2h5d1eGI=
|
||||
go.yaml.in/yaml/v2 v2.4.2/go.mod h1:081UH+NErpNdqlCXm3TtEran0rJZGxAYx9hb/ELlsPU=
|
||||
golang.org/x/net v0.43.0 h1:lat02VYK2j4aLzMzecihNvTlJNQUq316m2Mr9rnM6YE=
|
||||
golang.org/x/net v0.43.0/go.mod h1:vhO1fvI4dGsIjh73sWfUVjj3N7CA9WkKJNQm2svM6Jg=
|
||||
golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
|
||||
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
||||
golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
|
||||
golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
|
||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20241202173237-19429a94021a h1:hgh8P4EuoxpsuKMXX/To36nOFD7vixReXgn8lPGnt+o=
|
||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20241202173237-19429a94021a/go.mod h1:5uTbfoYQed2U9p3KIj2/Zzm02PYhndfdmML0qC3q3FU=
|
||||
google.golang.org/grpc v1.70.0 h1:pWFv03aZoHzlRKHWicjsZytKAiYCtNS0dHbXnIdq7jQ=
|
||||
google.golang.org/grpc v1.70.0/go.mod h1:ofIJqVKDXx/JiXrwr2IG4/zwdH9txy3IlF40RmcJSQw=
|
||||
google.golang.org/protobuf v1.36.8 h1:xHScyCOEuuwZEc6UtSOvPbAT4zRh0xcNRYekJwfqyMc=
|
||||
google.golang.org/protobuf v1.36.8/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
|
||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
81
internal/config/config.go
Normal file
81
internal/config/config.go
Normal file
@ -0,0 +1,81 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
ProfilePath string
|
||||
TasksWSURL string
|
||||
TasksAPIURL string
|
||||
TasksAgentToken string
|
||||
CoreEndpoint string
|
||||
CoreToken string
|
||||
CoreInsecure bool
|
||||
Model string
|
||||
MetricsListenAddr string
|
||||
HealthListenAddr string
|
||||
ShutdownTimeout time.Duration
|
||||
ResponseTimeout time.Duration
|
||||
}
|
||||
|
||||
func Load() (Config, error) {
|
||||
c := Config{
|
||||
ProfilePath: env("AGENT_PROFILE_PATH", "profiles/kaya.yaml"),
|
||||
TasksWSURL: os.Getenv("TASKS_WS_URL"),
|
||||
TasksAPIURL: os.Getenv("TASKS_API_URL"),
|
||||
TasksAgentToken: os.Getenv("TASKS_AGENT_TOKEN"),
|
||||
CoreEndpoint: env("CORE_ENDPOINT", "core:50051"),
|
||||
CoreToken: os.Getenv("CORE_TOKEN"),
|
||||
CoreInsecure: envBool("CORE_INSECURE", false),
|
||||
Model: env("MODEL", "deepseek-v4-flash"),
|
||||
MetricsListenAddr: env("METRICS_LISTEN_ADDR", ":9097"),
|
||||
HealthListenAddr: env("HEALTH_LISTEN_ADDR", ":8087"),
|
||||
ShutdownTimeout: envDur("SHUTDOWN_TIMEOUT", 15*time.Second),
|
||||
ResponseTimeout: envDur("RESPONSE_TIMEOUT", 120*time.Second),
|
||||
}
|
||||
if c.TasksWSURL == "" {
|
||||
return c, errors.New("TASKS_WS_URL is required")
|
||||
}
|
||||
if c.TasksAPIURL == "" {
|
||||
return c, errors.New("TASKS_API_URL is required")
|
||||
}
|
||||
if c.TasksAgentToken == "" {
|
||||
return c, errors.New("TASKS_AGENT_TOKEN is required")
|
||||
}
|
||||
if c.CoreToken == "" {
|
||||
return c, errors.New("CORE_TOKEN is required")
|
||||
}
|
||||
if c.ResponseTimeout <= 0 {
|
||||
return c, errors.New("RESPONSE_TIMEOUT must be positive")
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
|
||||
func env(k, d string) string {
|
||||
if v := os.Getenv(k); v != "" {
|
||||
return v
|
||||
}
|
||||
return d
|
||||
}
|
||||
|
||||
func envBool(k string, d bool) bool {
|
||||
if v := os.Getenv(k); v != "" {
|
||||
b, _ := strconv.ParseBool(v)
|
||||
return b
|
||||
}
|
||||
return d
|
||||
}
|
||||
|
||||
func envDur(k string, d time.Duration) time.Duration {
|
||||
if v := os.Getenv(k); v != "" {
|
||||
x, err := time.ParseDuration(v)
|
||||
if err == nil {
|
||||
return x
|
||||
}
|
||||
}
|
||||
return d
|
||||
}
|
||||
99
internal/inference/core.go
Normal file
99
internal/inference/core.go
Normal file
@ -0,0 +1,99 @@
|
||||
package inference
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/credentials"
|
||||
"google.golang.org/grpc/credentials/insecure"
|
||||
"google.golang.org/grpc/metadata"
|
||||
corepb "l-ai-agent-kaya/proto/l-ai-core-proto"
|
||||
)
|
||||
|
||||
type CoreRunner struct {
|
||||
api corepb.AIServiceClient
|
||||
token string
|
||||
conn *grpc.ClientConn
|
||||
}
|
||||
|
||||
func DialCore(target, token string, insecureTransport bool) (*CoreRunner, error) {
|
||||
var tc credentials.TransportCredentials
|
||||
if insecureTransport {
|
||||
tc = insecure.NewCredentials()
|
||||
} else {
|
||||
tc = credentials.NewTLS(&tls.Config{MinVersion: tls.VersionTLS12})
|
||||
}
|
||||
conn, err := grpc.NewClient(target, grpc.WithTransportCredentials(tc))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &CoreRunner{api: corepb.NewAIServiceClient(conn), token: token, conn: conn}, nil
|
||||
}
|
||||
|
||||
func (r *CoreRunner) Close() error {
|
||||
if r.conn != nil {
|
||||
return r.conn.Close()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *CoreRunner) auth(ctx context.Context) context.Context {
|
||||
if strings.TrimSpace(r.token) == "" {
|
||||
return ctx
|
||||
}
|
||||
return metadata.AppendToOutgoingContext(ctx, "authorization", "Bearer "+r.token)
|
||||
}
|
||||
|
||||
func (r *CoreRunner) RunAgent(ctx context.Context, req RunRequest) (RunResult, error) {
|
||||
msgs := make([]*corepb.StatelessMessage, 0, len(req.Messages))
|
||||
for _, m := range req.Messages {
|
||||
msgs = append(msgs, &corepb.StatelessMessage{Role: m.Role, Content: m.Content})
|
||||
}
|
||||
|
||||
toolsEnabled := req.ToolsEnabled
|
||||
maxIters := uint32(1)
|
||||
maxCalls := uint32(0)
|
||||
if toolsEnabled {
|
||||
maxIters = 12
|
||||
maxCalls = 20
|
||||
}
|
||||
|
||||
resp, err := r.api.RunAgent(r.auth(ctx), &corepb.AgentRequest{
|
||||
Messages: msgs,
|
||||
ModelName: req.Model,
|
||||
AgentId: req.AgentID,
|
||||
ToolLoop: &corepb.AgentToolLoopOptions{
|
||||
ToolsEnabled: toolsEnabled,
|
||||
MaxToolIterations: maxIters,
|
||||
MaxTotalToolCalls: maxCalls,
|
||||
TotalTimeoutSeconds: req.TimeoutSeconds,
|
||||
},
|
||||
ResponseContract: &corepb.AgentResponseContract{
|
||||
Format: corepb.AgentOutputFormat_AGENT_OUTPUT_FORMAT_TEXT,
|
||||
},
|
||||
RequestId: req.RequestID,
|
||||
IdempotencyKey: req.IdempotencyKey,
|
||||
SupportsDeferred: false,
|
||||
Metadata: req.Metadata,
|
||||
})
|
||||
if err != nil {
|
||||
return RunResult{}, err
|
||||
}
|
||||
if resp.Error != "" {
|
||||
return RunResult{}, fmt.Errorf("core RunAgent response error: %s", resp.Error)
|
||||
}
|
||||
if resp.Status == corepb.AnswerStatus_ANSWER_STATUS_PENDING {
|
||||
return RunResult{}, fmt.Errorf("unexpected deferred RunAgent response")
|
||||
}
|
||||
out := RunResult{Content: resp.Content, DurationMS: resp.DurationMs}
|
||||
if resp.Usage != nil {
|
||||
out.ToolsUsed = resp.Usage.ToolsUsed
|
||||
}
|
||||
if !toolsEnabled && len(out.ToolsUsed) != 0 {
|
||||
return RunResult{}, fmt.Errorf("tools used despite tools-disabled request")
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
29
internal/inference/types.go
Normal file
29
internal/inference/types.go
Normal file
@ -0,0 +1,29 @@
|
||||
package inference
|
||||
|
||||
import "context"
|
||||
|
||||
type ChatMessage struct {
|
||||
Role string
|
||||
Content string
|
||||
}
|
||||
|
||||
type RunRequest struct {
|
||||
Messages []ChatMessage
|
||||
Model string
|
||||
AgentID string
|
||||
RequestID string
|
||||
IdempotencyKey string
|
||||
TimeoutSeconds uint32
|
||||
Metadata map[string]string
|
||||
ToolsEnabled bool
|
||||
}
|
||||
|
||||
type RunResult struct {
|
||||
Content string
|
||||
DurationMS float64
|
||||
ToolsUsed []string
|
||||
}
|
||||
|
||||
type Runner interface {
|
||||
RunAgent(context.Context, RunRequest) (RunResult, error)
|
||||
}
|
||||
107
internal/profile/profile.go
Normal file
107
internal/profile/profile.go
Normal file
@ -0,0 +1,107 @@
|
||||
package profile
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
type Profile struct {
|
||||
Version int `yaml:"version"`
|
||||
Agent struct {
|
||||
ID string `yaml:"id"`
|
||||
Identity struct {
|
||||
Name string `yaml:"name"`
|
||||
Aliases map[string]string `yaml:"aliases"`
|
||||
Role string `yaml:"role"`
|
||||
SelfAwareness string `yaml:"self_awareness"`
|
||||
} `yaml:"identity"`
|
||||
PublicProfile struct {
|
||||
Bio string `yaml:"bio"`
|
||||
Status string `yaml:"status"`
|
||||
} `yaml:"public_profile"`
|
||||
} `yaml:"agent"`
|
||||
Personality struct {
|
||||
Traits []string `yaml:"traits"`
|
||||
Style struct {
|
||||
Tone string `yaml:"tone"`
|
||||
Verbosity string `yaml:"verbosity"`
|
||||
Formality string `yaml:"formality"`
|
||||
} `yaml:"style"`
|
||||
Avoid []string `yaml:"avoid"`
|
||||
} `yaml:"personality"`
|
||||
Behavior struct {
|
||||
Language struct {
|
||||
MatchConversation bool `yaml:"match_conversation"`
|
||||
RussianDisplayName string `yaml:"russian_display_name"`
|
||||
EnglishDisplayName string `yaml:"english_display_name"`
|
||||
} `yaml:"language"`
|
||||
} `yaml:"behavior"`
|
||||
LLM struct {
|
||||
Response struct {
|
||||
Model string `yaml:"model"`
|
||||
} `yaml:"response"`
|
||||
} `yaml:"llm"`
|
||||
}
|
||||
|
||||
func Load(path string) (Profile, error) {
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return Profile{}, err
|
||||
}
|
||||
var p Profile
|
||||
if err := yaml.Unmarshal(data, &p); err != nil {
|
||||
return Profile{}, err
|
||||
}
|
||||
if strings.TrimSpace(p.Agent.ID) == "" {
|
||||
return Profile{}, fmt.Errorf("profile %s: agent.id is required", path)
|
||||
}
|
||||
return p, nil
|
||||
}
|
||||
|
||||
// SystemPrompt builds a compact persona prompt for Tasks board mention replies.
|
||||
func (p Profile) SystemPrompt() string {
|
||||
name := p.Agent.Identity.Name
|
||||
if name == "" {
|
||||
name = p.Agent.ID
|
||||
}
|
||||
ru := p.Behavior.Language.RussianDisplayName
|
||||
if ru == "" {
|
||||
ru = p.Agent.Identity.Aliases["ru"]
|
||||
}
|
||||
var b strings.Builder
|
||||
b.WriteString(fmt.Sprintf("You are %s", name))
|
||||
if ru != "" && !strings.EqualFold(ru, name) {
|
||||
b.WriteString(fmt.Sprintf(" / %s", ru))
|
||||
}
|
||||
b.WriteString(fmt.Sprintf(" (agent_id=%s), an AI agent on the l-ai Tasks board.\n", p.Agent.ID))
|
||||
if bio := strings.TrimSpace(p.Agent.PublicProfile.Bio); bio != "" {
|
||||
b.WriteString(bio)
|
||||
b.WriteByte('\n')
|
||||
}
|
||||
if role := strings.TrimSpace(p.Agent.Identity.Role); role != "" {
|
||||
b.WriteString("Role: ")
|
||||
b.WriteString(role)
|
||||
b.WriteByte('\n')
|
||||
}
|
||||
if len(p.Personality.Traits) > 0 {
|
||||
b.WriteString("Traits: ")
|
||||
b.WriteString(strings.Join(p.Personality.Traits, ", "))
|
||||
b.WriteByte('\n')
|
||||
}
|
||||
if s := p.Personality.Style; s.Tone != "" || s.Verbosity != "" || s.Formality != "" {
|
||||
b.WriteString(fmt.Sprintf("Style: tone=%s verbosity=%s formality=%s\n", s.Tone, s.Verbosity, s.Formality))
|
||||
}
|
||||
if len(p.Personality.Avoid) > 0 {
|
||||
b.WriteString("Avoid: ")
|
||||
b.WriteString(strings.Join(p.Personality.Avoid, "; "))
|
||||
b.WriteByte('\n')
|
||||
}
|
||||
b.WriteString("You reply only to task comment @mentions. Be helpful and concise. ")
|
||||
b.WriteString("Match the language of the triggering comment (Russian or English). ")
|
||||
b.WriteString("Do not invent task state that is not in the provided context. ")
|
||||
b.WriteString("Output only the comment body to post — no preamble or quotes.\n")
|
||||
return b.String()
|
||||
}
|
||||
201
internal/tasksboard/loop.go
Normal file
201
internal/tasksboard/loop.go
Normal file
@ -0,0 +1,201 @@
|
||||
package tasksboard
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"l-ai-agent-kaya/internal/inference"
|
||||
"l-ai-agent-kaya/internal/profile"
|
||||
"l-ai-agent-kaya/internal/tasksws"
|
||||
)
|
||||
|
||||
const agentID = "kaya"
|
||||
|
||||
// Loop listens to Tasks board WS and replies to @kaya mentions.
|
||||
type Loop struct {
|
||||
WSURL string
|
||||
Token string
|
||||
REST *Client
|
||||
Core inference.Runner
|
||||
Profile profile.Profile
|
||||
Model string
|
||||
ResponseTimeout time.Duration
|
||||
Log *slog.Logger
|
||||
|
||||
mu sync.Mutex
|
||||
seen map[string]struct{}
|
||||
seenQ []string
|
||||
maxSeen int
|
||||
}
|
||||
|
||||
func (l *Loop) log() *slog.Logger {
|
||||
if l.Log != nil {
|
||||
return l.Log
|
||||
}
|
||||
return slog.Default()
|
||||
}
|
||||
|
||||
// Run blocks until ctx is cancelled.
|
||||
func (l *Loop) Run(ctx context.Context) {
|
||||
if l.seen == nil {
|
||||
l.seen = map[string]struct{}{}
|
||||
}
|
||||
if l.maxSeen <= 0 {
|
||||
l.maxSeen = 4096
|
||||
}
|
||||
client := &tasksws.Client{
|
||||
URL: l.WSURL,
|
||||
Token: l.Token,
|
||||
Agent: agentID,
|
||||
Log: l.log(),
|
||||
OnEvent: func(ctx context.Context, ev tasksws.Event) error {
|
||||
return l.handle(ctx, ev)
|
||||
},
|
||||
}
|
||||
l.log().Info("tasks board loop starting", "ws", l.WSURL, "agent", agentID)
|
||||
client.Run(ctx)
|
||||
}
|
||||
|
||||
func (l *Loop) handle(ctx context.Context, ev tasksws.Event) error {
|
||||
if ev.Type != "comment.added" {
|
||||
return nil
|
||||
}
|
||||
if ev.Comment == nil || ev.Task == nil {
|
||||
return nil
|
||||
}
|
||||
if !tasksws.MentionsSelf(ev, agentID) {
|
||||
return nil
|
||||
}
|
||||
if strings.EqualFold(ev.Comment.AuthorID, agentID) {
|
||||
return nil
|
||||
}
|
||||
commentID := ev.Comment.ID
|
||||
if commentID == "" {
|
||||
return nil
|
||||
}
|
||||
if !l.markSeen(commentID) {
|
||||
return nil
|
||||
}
|
||||
|
||||
taskID := ev.Task.ID
|
||||
l.log().Info("mention reply", "comment_id", commentID, "task_id", taskID, "author", ev.Comment.AuthorID)
|
||||
|
||||
replyCtx, cancel := context.WithTimeout(ctx, l.ResponseTimeout)
|
||||
defer cancel()
|
||||
|
||||
task, err := l.REST.GetTask(replyCtx, taskID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("get task: %w", err)
|
||||
}
|
||||
comments, err := l.REST.ListComments(replyCtx, taskID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("list comments: %w", err)
|
||||
}
|
||||
|
||||
userPrompt := buildUserPrompt(task, comments, commentID)
|
||||
timeoutSec := uint32(l.ResponseTimeout.Seconds())
|
||||
if timeoutSec < 30 {
|
||||
timeoutSec = 30
|
||||
}
|
||||
model := l.Model
|
||||
if model == "" && l.Profile.LLM.Response.Model != "" {
|
||||
model = l.Profile.LLM.Response.Model
|
||||
}
|
||||
if model == "" {
|
||||
model = "deepseek-v4-flash"
|
||||
}
|
||||
|
||||
res, err := l.Core.RunAgent(replyCtx, inference.RunRequest{
|
||||
Messages: []inference.ChatMessage{
|
||||
{Role: "system", Content: l.Profile.SystemPrompt()},
|
||||
{Role: "user", Content: userPrompt},
|
||||
},
|
||||
Model: model,
|
||||
AgentID: agentID,
|
||||
RequestID: "mention-reply:" + commentID,
|
||||
IdempotencyKey: "mention-reply:" + commentID,
|
||||
TimeoutSeconds: timeoutSec,
|
||||
Metadata: map[string]string{"trace_reason": "kaya_mention_reply", "comment_id": commentID, "task_id": taskID},
|
||||
ToolsEnabled: false,
|
||||
})
|
||||
if err != nil {
|
||||
return fmt.Errorf("RunAgent: %w", err)
|
||||
}
|
||||
body := strings.TrimSpace(res.Content)
|
||||
if body == "" {
|
||||
return fmt.Errorf("empty agent reply for comment %s", commentID)
|
||||
}
|
||||
|
||||
_, err = l.REST.PostComment(replyCtx, taskID, body, "mention-reply:"+commentID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("post comment: %w", err)
|
||||
}
|
||||
l.log().Info("mention reply posted", "comment_id", commentID, "task_id", taskID, "duration_ms", res.DurationMS)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (l *Loop) markSeen(id string) bool {
|
||||
l.mu.Lock()
|
||||
defer l.mu.Unlock()
|
||||
if _, ok := l.seen[id]; ok {
|
||||
return false
|
||||
}
|
||||
l.seen[id] = struct{}{}
|
||||
l.seenQ = append(l.seenQ, id)
|
||||
for len(l.seenQ) > l.maxSeen {
|
||||
old := l.seenQ[0]
|
||||
l.seenQ = l.seenQ[1:]
|
||||
delete(l.seen, old)
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func buildUserPrompt(task Task, comments []Comment, triggerID string) string {
|
||||
var b strings.Builder
|
||||
b.WriteString("Task context:\n")
|
||||
b.WriteString(fmt.Sprintf("- id: %s\n", task.ID))
|
||||
b.WriteString(fmt.Sprintf("- title: %s\n", task.Title))
|
||||
b.WriteString(fmt.Sprintf("- status: %s\n", task.Status))
|
||||
b.WriteString(fmt.Sprintf("- priority: %d\n", task.Priority))
|
||||
if task.Project != nil {
|
||||
b.WriteString(fmt.Sprintf("- project: %s (%s)\n", task.Project.Name, task.Project.Slug))
|
||||
}
|
||||
if task.Assignee != nil {
|
||||
name := task.Assignee.DisplayName
|
||||
if name == "" {
|
||||
name = task.Assignee.ID
|
||||
}
|
||||
b.WriteString(fmt.Sprintf("- assignee: %s/%s (%s)\n", task.Assignee.Type, task.Assignee.ID, name))
|
||||
}
|
||||
if desc := strings.TrimSpace(task.Description); desc != "" {
|
||||
b.WriteString("- description:\n")
|
||||
b.WriteString(trimRunes(desc, 4000))
|
||||
b.WriteByte('\n')
|
||||
}
|
||||
b.WriteString("\nComment thread (oldest first):\n")
|
||||
for _, c := range comments {
|
||||
marker := ""
|
||||
if c.ID == triggerID {
|
||||
marker = " ← reply to this"
|
||||
}
|
||||
b.WriteString(fmt.Sprintf("[%s] %s/%s:%s\n%s\n\n",
|
||||
c.CreatedAt.UTC().Format(time.RFC3339),
|
||||
c.AuthorType, c.AuthorID, marker,
|
||||
trimRunes(c.Body, 2000),
|
||||
))
|
||||
}
|
||||
b.WriteString("Write a helpful board comment reply to the marked mention.\n")
|
||||
return b.String()
|
||||
}
|
||||
|
||||
func trimRunes(s string, max int) string {
|
||||
r := []rune(s)
|
||||
if len(r) <= max {
|
||||
return s
|
||||
}
|
||||
return string(r[:max]) + "…"
|
||||
}
|
||||
119
internal/tasksboard/rest.go
Normal file
119
internal/tasksboard/rest.go
Normal file
@ -0,0 +1,119 @@
|
||||
package tasksboard
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Client struct {
|
||||
BaseURL string
|
||||
Token string
|
||||
HTTPClient *http.Client
|
||||
}
|
||||
|
||||
func (c *Client) http() *http.Client {
|
||||
if c.HTTPClient != nil {
|
||||
return c.HTTPClient
|
||||
}
|
||||
return http.DefaultClient
|
||||
}
|
||||
|
||||
type Task struct {
|
||||
ID string `json:"id"`
|
||||
Title string `json:"title"`
|
||||
Description string `json:"description"`
|
||||
Status string `json:"status"`
|
||||
Priority int `json:"priority"`
|
||||
Project *struct {
|
||||
ID string `json:"id"`
|
||||
Slug string `json:"slug"`
|
||||
Name string `json:"name"`
|
||||
} `json:"project,omitempty"`
|
||||
Assignee *struct {
|
||||
Type string `json:"type"`
|
||||
ID string `json:"id"`
|
||||
DisplayName string `json:"display_name,omitempty"`
|
||||
} `json:"assignee,omitempty"`
|
||||
}
|
||||
|
||||
type Comment struct {
|
||||
ID string `json:"id"`
|
||||
TaskID string `json:"task_id"`
|
||||
AuthorType string `json:"author_type"`
|
||||
AuthorID string `json:"author_id"`
|
||||
Body string `json:"body"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
}
|
||||
|
||||
func (c *Client) GetTask(ctx context.Context, taskID string) (Task, error) {
|
||||
var t Task
|
||||
if err := c.doJSON(ctx, http.MethodGet, "/api/v1/tasks/"+url.PathEscape(taskID), "", nil, &t); err != nil {
|
||||
return Task{}, err
|
||||
}
|
||||
return t, nil
|
||||
}
|
||||
|
||||
func (c *Client) ListComments(ctx context.Context, taskID string) ([]Comment, error) {
|
||||
var out struct {
|
||||
Comments []Comment `json:"comments"`
|
||||
}
|
||||
if err := c.doJSON(ctx, http.MethodGet, "/api/v1/tasks/"+url.PathEscape(taskID)+"/comments", "", nil, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out.Comments, nil
|
||||
}
|
||||
|
||||
func (c *Client) PostComment(ctx context.Context, taskID, body, idempotencyKey string) (Comment, error) {
|
||||
var out Comment
|
||||
payload := map[string]string{"body": body}
|
||||
if err := c.doJSON(ctx, http.MethodPost, "/api/v1/tasks/"+url.PathEscape(taskID)+"/comments", idempotencyKey, payload, &out); err != nil {
|
||||
return Comment{}, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *Client) doJSON(ctx context.Context, method, path, idempotencyKey string, body any, dst any) error {
|
||||
base := strings.TrimRight(c.BaseURL, "/")
|
||||
var rdr io.Reader
|
||||
if body != nil {
|
||||
b, err := json.Marshal(body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
rdr = bytes.NewReader(b)
|
||||
}
|
||||
req, err := http.NewRequestWithContext(ctx, method, base+path, rdr)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
req.Header.Set("Authorization", "Bearer "+c.Token)
|
||||
if body != nil {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
if idempotencyKey != "" {
|
||||
req.Header.Set("Idempotency-Key", idempotencyKey)
|
||||
}
|
||||
resp, err := c.http().Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
data, err := io.ReadAll(io.LimitReader(resp.Body, 1<<20))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||
return fmt.Errorf("tasks %s %s: status %d: %s", method, path, resp.StatusCode, strings.TrimSpace(string(data)))
|
||||
}
|
||||
if dst == nil || len(data) == 0 {
|
||||
return nil
|
||||
}
|
||||
return json.Unmarshal(data, dst)
|
||||
}
|
||||
185
internal/tasksws/client.go
Normal file
185
internal/tasksws/client.go
Normal file
@ -0,0 +1,185 @@
|
||||
// Package tasksws is a reconnecting WebSocket client for l-ai-tasks board events.
|
||||
package tasksws
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
// Event mirrors l-ai-tasks realtime.Event JSON.
|
||||
type Event struct {
|
||||
Type string `json:"type"`
|
||||
Seq int64 `json:"seq"`
|
||||
At time.Time `json:"at"`
|
||||
Project *Ref `json:"project,omitempty"`
|
||||
Task *TaskRef `json:"task,omitempty"`
|
||||
Comment *Comment `json:"comment,omitempty"`
|
||||
Mentions []Mention `json:"mentions,omitempty"`
|
||||
}
|
||||
|
||||
type Ref struct {
|
||||
ID string `json:"id"`
|
||||
Slug string `json:"slug"`
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type TaskRef struct {
|
||||
ID string `json:"id"`
|
||||
Title string `json:"title"`
|
||||
Status string `json:"status"`
|
||||
AssigneeID string `json:"assignee_id,omitempty"`
|
||||
}
|
||||
|
||||
type Comment struct {
|
||||
ID string `json:"id"`
|
||||
Body string `json:"body"`
|
||||
AuthorType string `json:"author_type"`
|
||||
AuthorID string `json:"author_id"`
|
||||
}
|
||||
|
||||
type Mention struct {
|
||||
Type string `json:"type"`
|
||||
ID string `json:"id"`
|
||||
}
|
||||
|
||||
// Handler is called for each board event (after ready).
|
||||
type Handler func(ctx context.Context, ev Event) error
|
||||
|
||||
// Client reconnects forever until ctx cancel.
|
||||
type Client struct {
|
||||
URL string
|
||||
Token string
|
||||
Agent string // logical id for filtering (nora, coordinator, kaya)
|
||||
Log *slog.Logger
|
||||
OnEvent Handler
|
||||
}
|
||||
|
||||
// Run blocks until ctx is done.
|
||||
func (c *Client) Run(ctx context.Context) {
|
||||
log := c.Log
|
||||
if log == nil {
|
||||
log = slog.Default()
|
||||
}
|
||||
var since int64
|
||||
attempt := 0
|
||||
for {
|
||||
if ctx.Err() != nil {
|
||||
return
|
||||
}
|
||||
err := c.session(ctx, &since)
|
||||
if ctx.Err() != nil {
|
||||
return
|
||||
}
|
||||
attempt++
|
||||
delay := backoff(attempt)
|
||||
log.Warn("tasks ws disconnected; reconnecting", "error", err, "attempt", attempt, "delay", delay, "since", since)
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-time.After(delay):
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) session(ctx context.Context, since *int64) error {
|
||||
u, err := url.Parse(c.URL)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
q := u.Query()
|
||||
q.Set("since", strconv.FormatInt(*since, 10))
|
||||
u.RawQuery = q.Encode()
|
||||
|
||||
hdr := http.Header{}
|
||||
if c.Token != "" {
|
||||
hdr.Set("Authorization", "Bearer "+c.Token)
|
||||
}
|
||||
dialer := websocket.Dialer{HandshakeTimeout: 15 * time.Second}
|
||||
conn, _, err := dialer.DialContext(ctx, u.String(), hdr)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
_ = conn.SetReadDeadline(time.Now().Add(90 * time.Second))
|
||||
conn.SetPongHandler(func(string) error {
|
||||
_ = conn.SetReadDeadline(time.Now().Add(90 * time.Second))
|
||||
return nil
|
||||
})
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return ctx.Err()
|
||||
default:
|
||||
}
|
||||
_, data, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_ = conn.SetReadDeadline(time.Now().Add(90 * time.Second))
|
||||
|
||||
var probe struct {
|
||||
Type string `json:"type"`
|
||||
Seq int64 `json:"seq"`
|
||||
}
|
||||
if err := json.Unmarshal(data, &probe); err != nil {
|
||||
continue
|
||||
}
|
||||
if probe.Type == "ready" {
|
||||
continue
|
||||
}
|
||||
var ev Event
|
||||
if err := json.Unmarshal(data, &ev); err != nil {
|
||||
continue
|
||||
}
|
||||
if ev.Seq > *since {
|
||||
*since = ev.Seq
|
||||
}
|
||||
if c.OnEvent != nil {
|
||||
if err := c.OnEvent(ctx, ev); err != nil && c.Log != nil {
|
||||
c.Log.Warn("tasks ws handler", "error", err, "type", ev.Type)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func backoff(attempt int) time.Duration {
|
||||
if attempt < 1 {
|
||||
attempt = 1
|
||||
}
|
||||
if attempt > 6 {
|
||||
attempt = 6
|
||||
}
|
||||
base := 200 * time.Millisecond
|
||||
d := base * time.Duration(1<<uint(attempt-1))
|
||||
if d > 5*time.Second {
|
||||
d = 5 * time.Second
|
||||
}
|
||||
jitter := time.Duration(rand.Int63n(int64(d/2) + 1))
|
||||
return d/2 + jitter
|
||||
}
|
||||
|
||||
// MentionsSelf reports whether ev.mentions includes agentID.
|
||||
func MentionsSelf(ev Event, agentID string) bool {
|
||||
for _, m := range ev.Mentions {
|
||||
if m.Type == "agent" && m.ID == agentID {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// ErrAuth is returned conceptually for logging; dial errors wrap status.
|
||||
func FormatDial(url, agent string) string {
|
||||
return fmt.Sprintf("tasksws agent=%s url=%s", agent, url)
|
||||
}
|
||||
48
profiles/kaya.yaml
Normal file
48
profiles/kaya.yaml
Normal file
@ -0,0 +1,48 @@
|
||||
version: 1
|
||||
|
||||
agent:
|
||||
id: kaya
|
||||
identity:
|
||||
name: Kaya
|
||||
aliases:
|
||||
ru: Кая
|
||||
role: BoardAssistant
|
||||
self_awareness: ai_agent
|
||||
public_profile:
|
||||
bio: "Friendly helper on the Tasks board — answers @kaya mentions in comments."
|
||||
status: Ready
|
||||
|
||||
personality:
|
||||
traits:
|
||||
- friendly
|
||||
- helpful
|
||||
- clear
|
||||
- concise
|
||||
- pragmatic
|
||||
- mildly_warm
|
||||
style:
|
||||
tone: friendly_helpful
|
||||
verbosity: low_to_medium
|
||||
formality: informal_professional
|
||||
avoid:
|
||||
- corporate_language
|
||||
- fake_certainty
|
||||
- excessive_disclaimers
|
||||
- long_unsolicited_essays
|
||||
- repeating_the_whole_thread
|
||||
|
||||
behavior:
|
||||
language:
|
||||
match_conversation: true
|
||||
russian_display_name: Кая
|
||||
english_display_name: Kaya
|
||||
board_comments:
|
||||
reply_to_mentions_only: true
|
||||
keep_replies_short: true
|
||||
|
||||
llm:
|
||||
response:
|
||||
model: deepseek-v4-flash
|
||||
|
||||
tools:
|
||||
enabled: false
|
||||
1854
proto/l-ai-core-proto/ai.pb.go
Normal file
1854
proto/l-ai-core-proto/ai.pb.go
Normal file
File diff suppressed because it is too large
Load Diff
243
proto/l-ai-core-proto/ai.proto
Normal file
243
proto/l-ai-core-proto/ai.proto
Normal file
@ -0,0 +1,243 @@
|
||||
syntax = "proto3";
|
||||
|
||||
package ai;
|
||||
|
||||
option go_package = "git.be.kaien.ru/l-ai/l-ai-core-proto/proto";
|
||||
|
||||
// AIRequest содержит запрос к AI модели
|
||||
message AIRequest {
|
||||
string text = 1; // Текст запроса
|
||||
string bot_username = 2; // Username бота для упоминаний
|
||||
string model_name = 3; // Название модели (по умолчанию "deepseek-v4-flash")
|
||||
bool enable_reasoning = 4; // Включить режим рассуждений (по умолчанию false)
|
||||
int64 global_user_id = 5; // Global user ID для получения доп. данных (профиль, воспоминания)
|
||||
AnswerGenerationOptions options = 6; // Дополнительные опции генерации ответа
|
||||
int64 telegram_user_id = 7; // Telegram user ID для применения пользовательских настроек (стиль ответа)
|
||||
int64 steam_user_id = 8; // SteamID64 отправителя для живых Steam-диалогов (l-ai-steam) - тот же "сигнал живого чата", что telegram_user_id, для второй платформы
|
||||
bool supports_deferred = 9; // Клиент умеет обрабатывать status=PENDING + GetAnswerJob (async тяжёлые тулзы). Старые клиенты не ставят — тяжёлые тулзы исполняются синхронно, как раньше.
|
||||
}
|
||||
|
||||
// AnswerGenerationOptions содержит дополнительные опции для генерации ответа
|
||||
message AnswerGenerationOptions {
|
||||
bool include_user_memories = 1; // Включить воспоминания о пользователе в промпт
|
||||
}
|
||||
|
||||
// AIResponse содержит ответ от AI модели
|
||||
message AIResponse {
|
||||
string content = 1; // Текст ответа (при status=PENDING — interim-сообщение)
|
||||
string error = 2; // Текст ошибки, если произошла
|
||||
double duration_ms = 3; // Время выполнения в миллисекундах
|
||||
string reasoning_content = 4; // Текст рассуждений (если включён режим reasoning)
|
||||
AnswerStatus status = 5; // COMPLETE (или UNSPECIFIED для старых клиентов) — content финальный;
|
||||
// PENDING — запущена долгая job, content это interim, финал тянуть через GetAnswerJob(job_id)
|
||||
string job_id = 6; // Идентификатор отложенной job (только при status=PENDING)
|
||||
}
|
||||
|
||||
// AnswerStatus описывает, готов ли ответ синхронно или отложен в job.
|
||||
enum AnswerStatus {
|
||||
ANSWER_STATUS_UNSPECIFIED = 0; // трактуется как COMPLETE (обратная совместимость)
|
||||
ANSWER_STATUS_COMPLETE = 1; // content — финальный ответ
|
||||
ANSWER_STATUS_PENDING = 2; // запущена долгая job; поллить GetAnswerJob(job_id)
|
||||
}
|
||||
|
||||
// GetAnswerJobRequest запрашивает статус отложенной генерации ответа.
|
||||
message GetAnswerJobRequest {
|
||||
string job_id = 1;
|
||||
}
|
||||
|
||||
// AnswerJobState — состояние отложенной генерации ответа.
|
||||
// Значения 0-4 исторические (GenerateAnswer deferred). Значения 5-7 добавлены для
|
||||
// durable RunAgent jobs; старые клиенты их не создают и потому не увидят. RUNNING
|
||||
// для старого клиента семантически эквивалентен PENDING (job ещё не терминальна).
|
||||
enum AnswerJobState {
|
||||
ANSWER_JOB_STATE_UNSPECIFIED = 0;
|
||||
ANSWER_JOB_STATE_PENDING = 1; // ещё не взята в работу / выполняется (evolver работает, модель дописывает)
|
||||
ANSWER_JOB_STATE_DONE = 2; // готово, content заполнен
|
||||
ANSWER_JOB_STATE_FAILED = 3; // ошибка, error заполнен
|
||||
ANSWER_JOB_STATE_NOT_FOUND = 4; // неизвестный/протухший job_id (напр. после рестарта core)
|
||||
ANSWER_JOB_STATE_RUNNING = 5; // взята воркером (lease активен), tool loop исполняется
|
||||
ANSWER_JOB_STATE_CANCELLED = 6; // отменена через CancelAnswerJob
|
||||
ANSWER_JOB_STATE_EXPIRED = 7; // истёк expiration до достижения терминального состояния
|
||||
}
|
||||
|
||||
// AnswerJobResponse — результат опроса отложенной генерации.
|
||||
message AnswerJobResponse {
|
||||
AnswerJobState state = 1;
|
||||
string content = 2; // финальный ответ, когда DONE
|
||||
string error = 3; // причина, когда FAILED
|
||||
double duration_ms = 4; // суммарное время генерации, когда DONE
|
||||
// Authoritative usage завершённой deferred RunAgent job — тот же счётчик, что
|
||||
// возвращает synchronous RunAgent (реальные iterations/tool calls core, не
|
||||
// self-reported из ответа модели). Для legacy GenerateAnswer jobs не заполняется.
|
||||
AgentUsage agent_usage = 5;
|
||||
}
|
||||
|
||||
// CreateSessionRequest для создания новой AI сессии
|
||||
message CreateSessionRequest {
|
||||
string user_id = 1; // Идентификатор пользователя
|
||||
string bot_username = 2; // Username бота для упоминаний
|
||||
string model_name = 3; // Название модели (по умолчанию "deepseek-v4-flash")
|
||||
int64 ttl_seconds = 4; // Время жизни сессии в секундах (по умолчанию 3600 = 1 час)
|
||||
}
|
||||
|
||||
// CreateSessionResponse содержит ID созданной сессии
|
||||
message CreateSessionResponse {
|
||||
string session_id = 1; // ID сессии
|
||||
string status = 2; // Статус сессии
|
||||
string error = 3; // Текст ошибки, если произошла
|
||||
}
|
||||
|
||||
// AskInSessionRequest для запроса в рамках сессии
|
||||
message AskInSessionRequest {
|
||||
string session_id = 1; // ID сессии
|
||||
string text = 2; // Текст запроса
|
||||
bool enable_reasoning = 3; // Включить режим рассуждений (по умолчанию false)
|
||||
int64 global_user_id = 4; // Global user ID для получения воспоминаний
|
||||
AnswerGenerationOptions options = 5; // Дополнительные опции генерации
|
||||
}
|
||||
|
||||
// SessionStatusResponse содержит статус сессии и ответ AI
|
||||
message SessionStatusResponse {
|
||||
string session_id = 1; // ID сессии
|
||||
string status = 2; // Статус сессии (active, expired, closed)
|
||||
string content = 3; // Текст ответа от AI
|
||||
string error = 4; // Текст ошибки, если произошла
|
||||
double duration_ms = 5; // Время выполнения в миллисекундах
|
||||
string reasoning_content = 6; // Текст рассуждений (если включён режим reasoning)
|
||||
}
|
||||
|
||||
// StatelessMessage одно сообщение для stateless запроса (role + content)
|
||||
message StatelessMessage {
|
||||
string role = 1; // "system", "user", "assistant"
|
||||
string content = 2; // Текст сообщения
|
||||
}
|
||||
|
||||
// StatelessRequest для stateless генерации с полным контролем над messages
|
||||
message StatelessRequest {
|
||||
repeated StatelessMessage messages = 1; // Массив сообщений (system + history + user)
|
||||
string model_name = 2; // Название модели (по умолчанию "deepseek-v4-flash")
|
||||
}
|
||||
|
||||
// HealthCheckRequest для проверки здоровья сервиса
|
||||
message HealthCheckRequest {}
|
||||
|
||||
// HealthCheckResponse для ответа на проверку здоровья
|
||||
message HealthCheckResponse {
|
||||
bool serving = 1;
|
||||
string version = 2;
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// RunAgent: надёжный агентный вызов (server-side tool filtering, deferred jobs).
|
||||
// В отличие от GenerateAnswer, НЕ подмешивает Telegram/Steam-персонализацию и не
|
||||
// требует telegram_user_id/steam_user_id. Вызывающая сторона полностью владеет
|
||||
// массивом messages (включая system). Требует service-to-service авторизации
|
||||
// (Bearer service-token в gRPC metadata) — agent_id/tool_profile сами по себе не
|
||||
// являются авторизацией.
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
// AgentToolLoopOptions задаёт лимиты tool loop. Значения клиента могут только
|
||||
// сузить серверные потолки (max_tool_iterations/max_total_tool_calls ≤ server max).
|
||||
message AgentToolLoopOptions {
|
||||
bool tools_enabled = 1; // false — модель отвечает без tools
|
||||
uint32 max_tool_iterations = 2; // потолок round-trip'ов модель↔tools (0 = серверный default)
|
||||
uint32 max_total_tool_calls = 3; // потолок РЕАЛЬНЫХ tool call'ов, не итераций (0 = серверный default)
|
||||
uint32 tool_timeout_seconds = 4; // таймаут одного tool call (0 = серверный default)
|
||||
uint32 total_timeout_seconds = 5; // wall-clock таймаут всего вызова (0 = серверный default), независим от budget
|
||||
}
|
||||
|
||||
// AgentOutputFormat — требуемый формат финального ответа.
|
||||
enum AgentOutputFormat {
|
||||
AGENT_OUTPUT_FORMAT_UNSPECIFIED = 0; // трактуется как TEXT
|
||||
AGENT_OUTPUT_FORMAT_TEXT = 1; // произвольный текст
|
||||
AGENT_OUTPUT_FORMAT_JSON = 2; // должен быть валидный JSON (без schema-проверки)
|
||||
AGENT_OUTPUT_FORMAT_JSON_SCHEMA = 3; // валидный JSON, проходящий json_schema
|
||||
}
|
||||
|
||||
// AgentResponseContract описывает требования к структурированному ответу.
|
||||
message AgentResponseContract {
|
||||
AgentOutputFormat format = 1;
|
||||
string json_schema = 2; // JSON Schema (draft-07), обязателен при format=JSON_SCHEMA
|
||||
bool repair_invalid_output = 3; // при провале валидации — один repair-запрос модели (без новых tools)
|
||||
}
|
||||
|
||||
// AgentRequest — запрос агентного вызова.
|
||||
message AgentRequest {
|
||||
repeated StatelessMessage messages = 1; // полный контекст, включая system
|
||||
string model_name = 2; // модель (по умолчанию deepseek-v4-flash)
|
||||
bool enable_reasoning = 3; // reasoning не попадает в обычный content
|
||||
|
||||
string agent_id = 4; // идентификатор логического агента (для логов/метрик, НЕ авторизация)
|
||||
string tool_profile = 5; // серверный профиль tools (НЕ авторизация; проверяется против credential)
|
||||
repeated string requested_tools = 6; // сужение профиля: effective = profile.allowed ∩ requested; пусто = все allowed
|
||||
|
||||
AgentToolLoopOptions tool_loop = 7;
|
||||
AgentResponseContract response_contract = 8;
|
||||
|
||||
string request_id = 9; // сквозной идентификатор для трассировки
|
||||
string idempotency_key = 10; // повторный RunAgent с тем же ключом не создаёт дубликат job
|
||||
bool supports_deferred = 11; // клиент умеет status=PENDING + GetAnswerJob для тяжёлых операций
|
||||
|
||||
map<string, string> metadata = 12; // произвольные метки (не логируются как секреты)
|
||||
}
|
||||
|
||||
// AgentUsage — счётчики использования за вызов.
|
||||
message AgentUsage {
|
||||
uint32 tool_iterations = 1; // сколько round-trip'ов модель↔tools произошло
|
||||
uint32 total_tool_calls = 2; // сколько РЕАЛЬНЫХ tool call'ов исполнено (3 calls в одной итерации = 3)
|
||||
repeated string tools_used = 3;
|
||||
uint64 input_tokens = 4;
|
||||
uint64 output_tokens = 5;
|
||||
}
|
||||
|
||||
// AgentResponse — результат агентного вызова.
|
||||
message AgentResponse {
|
||||
AnswerStatus status = 1; // COMPLETE — content финальный; PENDING — тянуть через GetAnswerJob(job_id)
|
||||
string content = 2;
|
||||
string error = 3;
|
||||
double duration_ms = 4;
|
||||
string job_id = 5; // при status=PENDING
|
||||
AgentUsage usage = 6;
|
||||
}
|
||||
|
||||
// CancelAnswerJobRequest — запрос отмены отложенной job (идемпотентный).
|
||||
message CancelAnswerJobRequest {
|
||||
string job_id = 1;
|
||||
string reason = 2; // причина отмены (сохраняется), опционально
|
||||
}
|
||||
|
||||
// CancelAnswerJobResponse — результат отмены.
|
||||
message CancelAnswerJobResponse {
|
||||
AnswerJobState state = 1; // состояние job после запроса (CANCELLED, либо терминальное, если уже завершилась)
|
||||
bool cancelled = 2; // true — отмена применена этим вызовом; false — job уже была терминальна/отменена
|
||||
string error = 3;
|
||||
}
|
||||
|
||||
// AIService - основной сервис для работы с AI
|
||||
service AIService {
|
||||
// RunAgent — надёжный агентный вызов с серверной фильтрацией tools и deferred
|
||||
// jobs. Требует service-to-service авторизации.
|
||||
rpc RunAgent(AgentRequest) returns (AgentResponse);
|
||||
|
||||
// CancelAnswerJob отменяет отложенную job (идемпотентно). Требует авторизации.
|
||||
rpc CancelAnswerJob(CancelAnswerJobRequest) returns (CancelAnswerJobResponse);
|
||||
|
||||
// GenerateAnswer генерирует ответ на запрос
|
||||
rpc GenerateAnswer(AIRequest) returns (AIResponse);
|
||||
|
||||
// GenerateStateless генерирует ответ из массива messages (system+history+user)
|
||||
rpc GenerateStateless(StatelessRequest) returns (AIResponse);
|
||||
|
||||
// HealthCheck проверяет здоровье сервиса
|
||||
rpc HealthCheck(HealthCheckRequest) returns (HealthCheckResponse);
|
||||
|
||||
// CreateSession созда новую AI сессию
|
||||
rpc CreateSession(CreateSessionRequest) returns (CreateSessionResponse);
|
||||
|
||||
// AskInSession отправляет запрос в рамках существующей сессии
|
||||
rpc AskInSession(AskInSessionRequest) returns (SessionStatusResponse);
|
||||
|
||||
// GetAnswerJob возвращает статус отложенной генерации ответа (когда GenerateAnswer
|
||||
// вернул status=PENDING). Клиент поллит его до DONE/FAILED и постит финал.
|
||||
rpc GetAnswerJob(GetAnswerJobRequest) returns (AnswerJobResponse);
|
||||
}
|
||||
411
proto/l-ai-core-proto/ai_grpc.pb.go
Normal file
411
proto/l-ai-core-proto/ai_grpc.pb.go
Normal file
@ -0,0 +1,411 @@
|
||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
||||
// versions:
|
||||
// - protoc-gen-go-grpc v1.6.1
|
||||
// - protoc v7.34.0
|
||||
// source: ai.proto
|
||||
|
||||
package proto
|
||||
|
||||
import (
|
||||
context "context"
|
||||
grpc "google.golang.org/grpc"
|
||||
codes "google.golang.org/grpc/codes"
|
||||
status "google.golang.org/grpc/status"
|
||||
)
|
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the grpc package it is being compiled against.
|
||||
// Requires gRPC-Go v1.64.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion9
|
||||
|
||||
const (
|
||||
AIService_RunAgent_FullMethodName = "/ai.AIService/RunAgent"
|
||||
AIService_CancelAnswerJob_FullMethodName = "/ai.AIService/CancelAnswerJob"
|
||||
AIService_GenerateAnswer_FullMethodName = "/ai.AIService/GenerateAnswer"
|
||||
AIService_GenerateStateless_FullMethodName = "/ai.AIService/GenerateStateless"
|
||||
AIService_HealthCheck_FullMethodName = "/ai.AIService/HealthCheck"
|
||||
AIService_CreateSession_FullMethodName = "/ai.AIService/CreateSession"
|
||||
AIService_AskInSession_FullMethodName = "/ai.AIService/AskInSession"
|
||||
AIService_GetAnswerJob_FullMethodName = "/ai.AIService/GetAnswerJob"
|
||||
)
|
||||
|
||||
// AIServiceClient is the client API for AIService service.
|
||||
//
|
||||
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
||||
//
|
||||
// AIService - основной сервис для работы с AI
|
||||
type AIServiceClient interface {
|
||||
// RunAgent — надёжный агентный вызов с серверной фильтрацией tools и deferred
|
||||
// jobs. Требует service-to-service авторизации.
|
||||
RunAgent(ctx context.Context, in *AgentRequest, opts ...grpc.CallOption) (*AgentResponse, error)
|
||||
// CancelAnswerJob отменяет отложенную job (идемпотентно). Требует авторизации.
|
||||
CancelAnswerJob(ctx context.Context, in *CancelAnswerJobRequest, opts ...grpc.CallOption) (*CancelAnswerJobResponse, error)
|
||||
// GenerateAnswer генерирует ответ на запрос
|
||||
GenerateAnswer(ctx context.Context, in *AIRequest, opts ...grpc.CallOption) (*AIResponse, error)
|
||||
// GenerateStateless генерирует ответ из массива messages (system+history+user)
|
||||
GenerateStateless(ctx context.Context, in *StatelessRequest, opts ...grpc.CallOption) (*AIResponse, error)
|
||||
// HealthCheck проверяет здоровье сервиса
|
||||
HealthCheck(ctx context.Context, in *HealthCheckRequest, opts ...grpc.CallOption) (*HealthCheckResponse, error)
|
||||
// CreateSession созда новую AI сессию
|
||||
CreateSession(ctx context.Context, in *CreateSessionRequest, opts ...grpc.CallOption) (*CreateSessionResponse, error)
|
||||
// AskInSession отправляет запрос в рамках существующей сессии
|
||||
AskInSession(ctx context.Context, in *AskInSessionRequest, opts ...grpc.CallOption) (*SessionStatusResponse, error)
|
||||
// GetAnswerJob возвращает статус отложенной генерации ответа (когда GenerateAnswer
|
||||
// вернул status=PENDING). Клиент поллит его до DONE/FAILED и постит финал.
|
||||
GetAnswerJob(ctx context.Context, in *GetAnswerJobRequest, opts ...grpc.CallOption) (*AnswerJobResponse, error)
|
||||
}
|
||||
|
||||
type aIServiceClient struct {
|
||||
cc grpc.ClientConnInterface
|
||||
}
|
||||
|
||||
func NewAIServiceClient(cc grpc.ClientConnInterface) AIServiceClient {
|
||||
return &aIServiceClient{cc}
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) RunAgent(ctx context.Context, in *AgentRequest, opts ...grpc.CallOption) (*AgentResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(AgentResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_RunAgent_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) CancelAnswerJob(ctx context.Context, in *CancelAnswerJobRequest, opts ...grpc.CallOption) (*CancelAnswerJobResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(CancelAnswerJobResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_CancelAnswerJob_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) GenerateAnswer(ctx context.Context, in *AIRequest, opts ...grpc.CallOption) (*AIResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(AIResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_GenerateAnswer_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) GenerateStateless(ctx context.Context, in *StatelessRequest, opts ...grpc.CallOption) (*AIResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(AIResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_GenerateStateless_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) HealthCheck(ctx context.Context, in *HealthCheckRequest, opts ...grpc.CallOption) (*HealthCheckResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(HealthCheckResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_HealthCheck_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) CreateSession(ctx context.Context, in *CreateSessionRequest, opts ...grpc.CallOption) (*CreateSessionResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(CreateSessionResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_CreateSession_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) AskInSession(ctx context.Context, in *AskInSessionRequest, opts ...grpc.CallOption) (*SessionStatusResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(SessionStatusResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_AskInSession_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *aIServiceClient) GetAnswerJob(ctx context.Context, in *GetAnswerJobRequest, opts ...grpc.CallOption) (*AnswerJobResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(AnswerJobResponse)
|
||||
err := c.cc.Invoke(ctx, AIService_GetAnswerJob_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// AIServiceServer is the server API for AIService service.
|
||||
// All implementations must embed UnimplementedAIServiceServer
|
||||
// for forward compatibility.
|
||||
//
|
||||
// AIService - основной сервис для работы с AI
|
||||
type AIServiceServer interface {
|
||||
// RunAgent — надёжный агентный вызов с серверной фильтрацией tools и deferred
|
||||
// jobs. Требует service-to-service авторизации.
|
||||
RunAgent(context.Context, *AgentRequest) (*AgentResponse, error)
|
||||
// CancelAnswerJob отменяет отложенную job (идемпотентно). Требует авторизации.
|
||||
CancelAnswerJob(context.Context, *CancelAnswerJobRequest) (*CancelAnswerJobResponse, error)
|
||||
// GenerateAnswer генерирует ответ на запрос
|
||||
GenerateAnswer(context.Context, *AIRequest) (*AIResponse, error)
|
||||
// GenerateStateless генерирует ответ из массива messages (system+history+user)
|
||||
GenerateStateless(context.Context, *StatelessRequest) (*AIResponse, error)
|
||||
// HealthCheck проверяет здоровье сервиса
|
||||
HealthCheck(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error)
|
||||
// CreateSession созда новую AI сессию
|
||||
CreateSession(context.Context, *CreateSessionRequest) (*CreateSessionResponse, error)
|
||||
// AskInSession отправляет запрос в рамках существующей сессии
|
||||
AskInSession(context.Context, *AskInSessionRequest) (*SessionStatusResponse, error)
|
||||
// GetAnswerJob возвращает статус отложенной генерации ответа (когда GenerateAnswer
|
||||
// вернул status=PENDING). Клиент поллит его до DONE/FAILED и постит финал.
|
||||
GetAnswerJob(context.Context, *GetAnswerJobRequest) (*AnswerJobResponse, error)
|
||||
mustEmbedUnimplementedAIServiceServer()
|
||||
}
|
||||
|
||||
// UnimplementedAIServiceServer must be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedAIServiceServer struct{}
|
||||
|
||||
func (UnimplementedAIServiceServer) RunAgent(context.Context, *AgentRequest) (*AgentResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method RunAgent not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) CancelAnswerJob(context.Context, *CancelAnswerJobRequest) (*CancelAnswerJobResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method CancelAnswerJob not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) GenerateAnswer(context.Context, *AIRequest) (*AIResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method GenerateAnswer not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) GenerateStateless(context.Context, *StatelessRequest) (*AIResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method GenerateStateless not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) HealthCheck(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method HealthCheck not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) CreateSession(context.Context, *CreateSessionRequest) (*CreateSessionResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method CreateSession not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) AskInSession(context.Context, *AskInSessionRequest) (*SessionStatusResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method AskInSession not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) GetAnswerJob(context.Context, *GetAnswerJobRequest) (*AnswerJobResponse, error) {
|
||||
return nil, status.Error(codes.Unimplemented, "method GetAnswerJob not implemented")
|
||||
}
|
||||
func (UnimplementedAIServiceServer) mustEmbedUnimplementedAIServiceServer() {}
|
||||
func (UnimplementedAIServiceServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeAIServiceServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to AIServiceServer will
|
||||
// result in compilation errors.
|
||||
type UnsafeAIServiceServer interface {
|
||||
mustEmbedUnimplementedAIServiceServer()
|
||||
}
|
||||
|
||||
func RegisterAIServiceServer(s grpc.ServiceRegistrar, srv AIServiceServer) {
|
||||
// If the following call panics, it indicates UnimplementedAIServiceServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&AIService_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
func _AIService_RunAgent_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(AgentRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).RunAgent(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_RunAgent_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).RunAgent(ctx, req.(*AgentRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_CancelAnswerJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(CancelAnswerJobRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).CancelAnswerJob(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_CancelAnswerJob_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).CancelAnswerJob(ctx, req.(*CancelAnswerJobRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_GenerateAnswer_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(AIRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).GenerateAnswer(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_GenerateAnswer_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).GenerateAnswer(ctx, req.(*AIRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_GenerateStateless_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(StatelessRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).GenerateStateless(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_GenerateStateless_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).GenerateStateless(ctx, req.(*StatelessRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_HealthCheck_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(HealthCheckRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).HealthCheck(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_HealthCheck_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).HealthCheck(ctx, req.(*HealthCheckRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_CreateSession_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(CreateSessionRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).CreateSession(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_CreateSession_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).CreateSession(ctx, req.(*CreateSessionRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_AskInSession_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(AskInSessionRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).AskInSession(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_AskInSession_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).AskInSession(ctx, req.(*AskInSessionRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _AIService_GetAnswerJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(GetAnswerJobRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(AIServiceServer).GetAnswerJob(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: AIService_GetAnswerJob_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(AIServiceServer).GetAnswerJob(ctx, req.(*GetAnswerJobRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
// AIService_ServiceDesc is the grpc.ServiceDesc for AIService service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
// and not to be introspected or modified (even as a copy)
|
||||
var AIService_ServiceDesc = grpc.ServiceDesc{
|
||||
ServiceName: "ai.AIService",
|
||||
HandlerType: (*AIServiceServer)(nil),
|
||||
Methods: []grpc.MethodDesc{
|
||||
{
|
||||
MethodName: "RunAgent",
|
||||
Handler: _AIService_RunAgent_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "CancelAnswerJob",
|
||||
Handler: _AIService_CancelAnswerJob_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "GenerateAnswer",
|
||||
Handler: _AIService_GenerateAnswer_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "GenerateStateless",
|
||||
Handler: _AIService_GenerateStateless_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "HealthCheck",
|
||||
Handler: _AIService_HealthCheck_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "CreateSession",
|
||||
Handler: _AIService_CreateSession_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "AskInSession",
|
||||
Handler: _AIService_AskInSession_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "GetAnswerJob",
|
||||
Handler: _AIService_GetAnswerJob_Handler,
|
||||
},
|
||||
},
|
||||
Streams: []grpc.StreamDesc{},
|
||||
Metadata: "ai.proto",
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user