function calling begin

Author: saintbyte

constants.go

@@ -1,18 +1,43 @@
 package gigachat
 
+// Авторизауия
 const (
-	GigaChatTokenFileEnv           = "GIGACHAT_TOKEN_FILE"   // Перемеменная среды с путем к файлу с токеном
-	GigaChatExpiresFileEnv         = "GIGACHAT_EXPIRES_FILE" // Переменная среды с путем к файл где время устревания токена
-	GigaChatToken                  = "GIGACHAT_TOKEN"        // Или токен берем из окружения
-	GigaChatAuthData               = "GIGACHAT_AUTH_DATA"    // Данные дла авторизации чтоб получить токен
-	GigaChatOauthUrl               = "https://ngw.devices.sberbank.ru:9443/api/v2/oauth"
-	GigaChatApiHost                = "gigachat.devices.sberbank.ru" //1 - gigachat.devices.sberbank.ru 2  gigachat-preview.devices.sberbank.ru
-	GigaChatModelsPath             = "/api/v1/models"
-	GigaChatChatCompletionPath     = "/api/v1/chat/completions"
-	GigaChatEmbeddingsPath         = "/api/v1/embeddings"
-	GigaChatModel                  = "GigaChat"
-	GigaChatMaxTokens              = 16384
-	MaxEmbeddingSize           int = 8192
-	GigaChatRoleUser               = "user"
-	GigaChatRoleSystem             = "system"
+	GigaChatTokenFileEnv   = "GIGACHAT_TOKEN_FILE"   // Перемеменная среды с путем к файлу с токеном
+	GigaChatExpiresFileEnv = "GIGACHAT_EXPIRES_FILE" // Переменная среды с путем к файл где время устревания токена
+	GigaChatToken          = "GIGACHAT_TOKEN"        // Или токен берем из окружения
+	GigaChatAuthData       = "GIGACHAT_AUTH_DATA"    // Данные дла авторизации чтоб получить токен
+	GigaChatOauthUrl       = "https://ngw.devices.sberbank.ru:9443/api/v2/oauth"
 )
+
+// Ручки API
+const (
+	GigaChatApiHost            = "gigachat.devices.sberbank.ru" //1 - gigachat.devices.sberbank.ru 2  gigachat-preview.devices.sberbank.ru
+	GigaChatModelsPath         = "/api/v1/models"
+	GigaChatChatCompletionPath = "/api/v1/chat/completions"
+	GigaChatEmbeddingsPath     = "/api/v1/embeddings"
+)
+
+// Настройки
+const (
+	GigaChatModel         = "GigaChat" // GigaChat, GigaChat-Pro, GigaChat-Max Если тестовый хост то GigaChat-Pro-preview
+	GigaChatMaxTokens     = 16384
+	MaxEmbeddingSize      = 8192
+	GigaChatRoleUser      = "user"
+	GigaChatRoleSystem    = "system"
+	GigaChatRoleAssistant = "assistant"
+)
+
+//'finish_reason': 'function_call'
+/*
+finish_reason
+string
+Возможные значения: [stop, length, function_call, blacklist, error]
+
+Причина завершения гипотезы. Возможные значения:
+
+stop — модель закончила формировать гипотезу и вернула полный ответ;
+length — достигнут лимит токенов в сообщении;
+function_call — указывает, что при запросе была вызвана встроенная функция или сгенерированы аргументы для пользовательской функции;
+blacklist — запрос попадает под тематические ограничения.
+error — ответ модели содержит невалидные аргументы пользовательской функции.
+*/

structures.go

@@ -26,9 +26,10 @@ type MessageRequest struct {
 
 // Статистика использования
 type Usage struct {
-	PromptTokens     int `json:"prompt_tokens"`
-	CompletionTokens int `json:"completion_tokens,omitempty"`
-	TotalTokens      int `json:"total_tokens,omitempty"`
+	// Данные об использовании модели.
+	PromptTokens     int `json:"prompt_tokens"`               // int32 Количество токенов во входящем сообщении (роль user).
+	CompletionTokens int `json:"completion_tokens,omitempty"` // int32  Количество токенов, сгенерированных моделью (роль assistant).
+	TotalTokens      int `json:"total_tokens,omitempty"`      //int32 Общее количество токенов.
 }
 
 type MessageResponse struct {