Skip to content

Commit

Permalink
fix: update gititgnore
Browse files Browse the repository at this point in the history
Signed-off-by: Zhou Zhiqiang <[email protected]>
  • Loading branch information
STRRL committed Mar 19, 2024
1 parent f0e51d6 commit fa5fe22
Show file tree
Hide file tree
Showing 2 changed files with 43 additions and 1 deletion.
1 change: 0 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -17,4 +17,3 @@ out/
*.tiktoken

config.json
unollm
43 changes: 43 additions & 0 deletions cmd/unollm/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
package main

import (
"context"
"log"
"log/slog"
"net"
"os"

"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/logging"
"go.limit.dev/unollm/grpcServer"
"go.limit.dev/unollm/model"
"google.golang.org/grpc"
"google.golang.org/grpc/reflection"
)

func InterceptorLogger(l *slog.Logger) logging.Logger {
return logging.LoggerFunc(func(ctx context.Context, lvl logging.Level, msg string, fields ...any) {
l.Log(ctx, slog.Level(lvl), msg, fields...)
})
}

func main() {
logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
opts := []logging.Option{
logging.WithLogOnEvents(logging.StartCall, logging.FinishCall),
}

// start grpc server
svr := grpc.NewServer(grpc.ChainUnaryInterceptor(
logging.UnaryServerInterceptor(InterceptorLogger(logger), opts...),
))

reflection.Register(svr)
logger.Info("start grpc server", "addr", "0.0.0.0:19198")
tcpList, err := net.Listen("tcp", "0.0.0.0:19198")
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
service := grpcServer.UnoForwardServer{}
model.RegisterUnoLLMv1Server(svr, &service)
svr.Serve(tcpList)
}

0 comments on commit fa5fe22

Please sign in to comment.