Реализована операции Milvus для управления документами и встраиванием, включая функции вставки, запроса и удаления. Внедрите архитектуру RAG с LLM и сервисами встраивания. Добавьте обработку текста для фрагментации и конкатенации. Создайте автономный скрипт для настройки и управления Milvus. Разработайте комплексные тесты API для обработки документов и взаимодействия с LLM, включая имитации для сервисов. Расширьте возможности конфигурации пользователя с помощью дополнительных настроек YAML.

tests/openrouter_test.go

@@ -0,0 +1,125 @@
+package api_test
+
+import (
+	"context"
+	openroute2 "easy_rag/internal/llm/openroute"
+	"fmt"
+	"testing"
+)
+
+func TestFetchChatCompletions(t *testing.T) {
+	client := openroute2.NewOpenRouterClient("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")
+
+	request := openroute2.Request{
+		Model: "qwen/qwen2.5-vl-72b-instruct:free",
+		Messages: []openroute2.MessageRequest{
+			{openroute2.RoleUser, "Привет!", "", ""},
+		},
+	}
+
+	output, err := client.FetchChatCompletions(request)
+	if err != nil {
+		t.Errorf("error %v", err)
+	}
+
+	t.Logf("output: %v", output.Choices[0].Message.Content)
+}
+
+func TestFetchChatCompletionsStreaming(t *testing.T) {
+	client := openroute2.NewOpenRouterClient("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")
+
+	request := openroute2.Request{
+		Model: "qwen/qwen2.5-vl-72b-instruct:free",
+		Messages: []openroute2.MessageRequest{
+			{openroute2.RoleUser, "Привет!", "", ""},
+		},
+		Stream: true,
+	}
+
+	outputChan := make(chan openroute2.Response)
+	processingChan := make(chan interface{})
+	errChan := make(chan error)
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	go client.FetchChatCompletionsStream(request, outputChan, processingChan, errChan, ctx)
+
+	for {
+		select {
+		case output := <-outputChan:
+			if len(output.Choices) > 0 {
+				t.Logf("%s", output.Choices[0].Delta.Content)
+			}
+		case <-processingChan:
+			t.Logf("Обработка\n")
+		case err := <-errChan:
+			if err != nil {
+				t.Errorf("Ошибка: %v", err)
+				return
+			}
+			return
+		case <-ctx.Done():
+			fmt.Println("Контекст отменен:", ctx.Err())
+			return
+		}
+	}
+}
+
+func TestFetchChatCompletionsAgentStreaming(t *testing.T) {
+	client := openroute2.NewOpenRouterClient("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")
+	agent := openroute2.NewRouterAgent(client, "qwen/qwen2.5-vl-72b-instruct:freet", openroute2.RouterAgentConfig{
+		Temperature: 0.7,
+		MaxTokens:   100,
+	})
+
+	outputChan := make(chan openroute2.Response)
+	processingChan := make(chan interface{})
+	errChan := make(chan error)
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	chat := []openroute2.MessageRequest{
+		{Role: openroute2.RoleSystem, Content: "Вы полезный помощник."},
+		{Role: openroute2.RoleUser, Content: "Привет!"},
+	}
+
+	go agent.ChatStream(chat, outputChan, processingChan, errChan, ctx)
+
+	for {
+		select {
+		case output := <-outputChan:
+			if len(output.Choices) > 0 {
+				t.Logf("%s", output.Choices[0].Delta.Content)
+			}
+		case <-processingChan:
+			t.Logf("Обработка\n")
+		case err := <-errChan:
+			if err != nil {
+				t.Errorf("Ошибка: %v", err)
+				return
+			}
+			return
+		case <-ctx.Done():
+			fmt.Println("Контекст отменен:", ctx.Err())
+			return
+		}
+	}
+}
+
+func TestFetchChatCompletionsAgentSimpleChat(t *testing.T) {
+	client := openroute2.NewOpenRouterClient("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")
+	agent := openroute2.NewRouterAgentChat(client, "qwen/qwen2.5-vl-72b-instruct:free", openroute2.RouterAgentConfig{
+		Temperature: 0.0,
+		MaxTokens:   100,
+	}, "Вы полезный помощник, отвечайте короткими словами.")
+
+	agent.Chat("Запомни это: \"wojtess\"")
+	agent.Chat("Что я просил вас запомнить?")
+
+	for _, msg := range agent.Messages {
+		content, ok := msg.Content.(string)
+		if ok {
+			t.Logf("%s: %s", msg.Role, content)
+		}
+	}
+}