/
githubmirror
/
Multi-GPT
Обзор
Документация
Войти
/
githubmirror
/
Multi-GPT
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/integration/milvus_memory_tests.py
48 строк
1 KB
xuqi.wxq
Fix Milvus as a long-term memory backend.
15 апр 2023, 20:45
15 апр 2023, 20:45
5e189c8
Код
Авторство
О чём код?
import random import string import unittest from autogpt.config import Config from autogpt.memory.milvus import MilvusMemory class TestMilvusMemory(unittest.TestCase): def random_string(self, length): return "".join(random.choice(string.ascii_letters) for _ in range(length)) def setUp(self): cfg = Config() cfg.milvus_addr = "localhost:19530" self.memory = MilvusMemory(cfg) self.memory.clear() # Add example texts to the cache self.example_texts = [ "The quick brown fox jumps over the lazy dog", "I love machine learning and natural language processing", "The cake is a lie, but the pie is always true", "ChatGPT is an advanced AI model for conversation", ] for text in self.example_texts: self.memory.add(text) # Add some random strings to test noise for _ in range(5): self.memory.add(self.random_string(10)) def test_get_relevant(self): query = "I'm interested in artificial intelligence and NLP" k = 3 relevant_texts = self.memory.get_relevant(query, k) print(f"Top {k} relevant texts for the query '{query}':") for i, text in enumerate(relevant_texts, start=1): print(f"{i}. {text}") self.assertEqual(len(relevant_texts), k) self.assertIn(self.example_texts[1], relevant_texts) if __name__ == "__main__": unittest.main()