Nik Afiq
ad50d641bd
- Updated LLMClient interface to support model-specific generation and model listing. - Integrated model store and validator into the command application for managing AI models. - Implemented commands for setting, getting, and listing active AI models in Discord. - Enhanced AI query handling to utilize the selected model and return model information in responses. - Added caching mechanism for model validation to improve performance. - Introduced gRPC methods for listing available AI models in the ai-gateway. - Updated protobuf definitions to include model-related fields and messages. - Added tests for model store and validator functionalities.
30 lines
545 B
Protocol Buffer
30 lines
545 B
Protocol Buffer
syntax = "proto3";
|
|
|
|
package ai.v1;
|
|
|
|
option go_package = "gitea.nik4nao.com/nik/home-services/gen/ai/v1;aiv1";
|
|
|
|
service AIService {
|
|
rpc Query(QueryRequest) returns (QueryResponse);
|
|
rpc ListModels(ListModelsRequest) returns (ListModelsResponse);
|
|
}
|
|
|
|
message QueryRequest {
|
|
string text = 1;
|
|
string source = 2;
|
|
string model = 3;
|
|
}
|
|
|
|
message QueryResponse {
|
|
string reply = 1;
|
|
string intent = 2;
|
|
bool action_taken = 3;
|
|
string model_used = 4;
|
|
}
|
|
|
|
message ListModelsRequest {}
|
|
|
|
message ListModelsResponse {
|
|
repeated string names = 1;
|
|
}
|