Add files via upload

ulab-uiuc · May 4, 2024 · eca4da2 · eca4da2
1 parent 30eb038
commit eca4da2
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/code_test.py b/tests/code_test.py
@@ -0,0 +1,11 @@
+from agent_base import *
+from kb_base import *
+base_agent=BaseResearchAgent('Jiaxuan You')
+# print(base_agent.profile)
+
+kb=BaseKnowledgeBase()
+# print(kb.get_data(10,"Machine Learning"))
+data=kb.get_data(10,"Machine Learning")
+paper=base_agent.read_paper(data,"Machine Learning")
+idea=base_agent.generate_idea(data,"Machine Learning")
+relations=base_agent.get_co_author_relationships('Jiaxuan You',3)