gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge remote-tracking branch 'origin/update_tests' into fix/sdk-and-config

Browse source
This commit is contained in:
Boris Arzentar 2024-06-03 14:55:58 +02:00
commit cbf701db6c
2 changed files with 36 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
.gitignore vendored
View file

@ -1,5 +1,9 @@
.data
.env
.local.env
.prod.env
cognee/.data/
# Byte-compiled / optimized / DLL files
__pycache__/

32
cognee/tests/test_library.py
View file

@ -45,6 +45,31 @@ async def main():
Some notable LLMs are OpenAI's GPT series of models (e.g., GPT-3.5 and GPT-4, used in ChatGPT and Microsoft Copilot), Google's PaLM and Gemini (the latter of which is currently used in the chatbot of the same name), xAI's Grok, Meta's LLaMA family of open-source models, Anthropic's Claude models, Mistral AI's open source models, and Databricks' open source DBRX.
"""
await cognee.cognify([ "cs_explanations"])
search_results = cognee.search("SIMILARITY", "computer science")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("CATEGORIES", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("NEIGHBOR", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("SUMMARY", "Work and computers")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("ADJACENT", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
#
# dataset_name = "cs_explanations"
# await cognee.add(
@ -55,7 +80,12 @@ async def main():
# dataset_name
# )
await cognee.cognify([ "cs_explanations"])
# search_results = cognee.search("SIMILARITY", "computer science")
# if len(search_results) == 0:
# print("The similarity search results list is empty.")
# raise Exception("The search results list is empty.")
# else:
# print("The search results list is not empty.")
if __name__ == "__main__":