raznis commited on
Commit
1e78c7b
·
verified ·
1 Parent(s): 71aa48c

Upload folder using huggingface_hub

Browse files
.gitignore CHANGED
@@ -177,3 +177,5 @@ cython_debug/
177
  output/*
178
  investigators/output/*
179
  search_results_2025*
 
 
 
177
  output/*
178
  investigators/output/*
179
  search_results_2025*
180
+ .DS_Store
181
+ memory/*
app.py CHANGED
@@ -58,15 +58,18 @@ with gr.Blocks() as view:
58
  outputs=output
59
  )
60
 
61
-
62
-
63
  clear_btn.click(
64
  fn=clear_inputs,
65
  inputs=[],
66
  outputs=[name_input, affiliation_input]
67
  )
68
  gr.Examples(
69
- examples=[["Raz Nissim", "Ben Gurion University, General Motors"]],
 
 
 
 
 
70
  inputs=[name_input, affiliation_input]
71
  )
72
 
 
58
  outputs=output
59
  )
60
 
 
 
61
  clear_btn.click(
62
  fn=clear_inputs,
63
  inputs=[],
64
  outputs=[name_input, affiliation_input]
65
  )
66
  gr.Examples(
67
+ examples=[["Raz Nissim", "Ben Gurion University, General Motors"],
68
+ ["Mohammed Mosharref Hossain", "Albany"],
69
+ ["Giovanni Cazzetta", "Montreal"],
70
+ ["Willy Bokonga", "Congo"],
71
+ ["Avraham Hirshzon", "Israel, Politician"],
72
+ ],
73
  inputs=[name_input, affiliation_input]
74
  )
75
 
investigators/src/investigators/config/agents.yaml CHANGED
@@ -9,6 +9,7 @@ researcher:
9
  finding and connecting information about people and businesses. You know how to
10
  follow information trails and identify valuable data sources.
11
  You have the ability to search the web using search tools.
 
12
  llm: openai/gpt-4o
13
 
14
  fincrime_analyst:
 
9
  finding and connecting information about people and businesses. You know how to
10
  follow information trails and identify valuable data sources.
11
  You have the ability to search the web using search tools.
12
+ You have the ability to use your memory to gather information.
13
  llm: openai/gpt-4o
14
 
15
  fincrime_analyst:
investigators/src/investigators/crew.py CHANGED
@@ -5,6 +5,11 @@ from crewai import TaskOutput
5
  from crewai_tools import SerperDevTool
6
  from crewai_tools import ScrapeWebsiteTool
7
  from crewai_tools import BraveSearchTool
 
 
 
 
 
8
  NO_INFORMATION_FOUND="no information found"
9
 
10
  def validate_researcher_content(result: TaskOutput) -> Tuple[bool, Any]:
@@ -93,4 +98,24 @@ class Investigators():
93
  tasks=self.tasks, # Automatically created by the @task decorator
94
  process=Process.sequential,
95
  verbose=True,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
96
  )
 
5
  from crewai_tools import SerperDevTool
6
  from crewai_tools import ScrapeWebsiteTool
7
  from crewai_tools import BraveSearchTool
8
+
9
+ from crewai.memory import LongTermMemory, EntityMemory
10
+ from crewai.memory.storage.rag_storage import RAGStorage
11
+ from crewai.memory.storage.ltm_sqlite_storage import LTMSQLiteStorage
12
+
13
  NO_INFORMATION_FOUND="no information found"
14
 
15
  def validate_researcher_content(result: TaskOutput) -> Tuple[bool, Any]:
 
98
  tasks=self.tasks, # Automatically created by the @task decorator
99
  process=Process.sequential,
100
  verbose=True,
101
+ memory=True,
102
+ # Long-term memory for persistent storage across sessions
103
+ long_term_memory = LongTermMemory(
104
+ storage=LTMSQLiteStorage(
105
+ db_path="./memory/long_term_memory_storage.db"
106
+ )
107
+ ),
108
+ # Entity memory for tracking key information about entities
109
+ entity_memory = EntityMemory(
110
+ storage=RAGStorage(
111
+ embedder_config={
112
+ "provider": "openai",
113
+ "config": {
114
+ "model": 'text-embedding-3-small'
115
+ }
116
+ },
117
+ type="short_term",
118
+ path="./memory/"
119
+ )
120
+ ),
121
  )