dwb2023 commited on
Commit
e8a087b
·
1 Parent(s): d0ae247

ensuree rows returned from sqlite are dictionary objects not tuples

Browse files
Files changed (2) hide show
  1. src/relationships.py +2 -1
  2. ui/format.py +4 -3
src/relationships.py CHANGED
@@ -29,9 +29,10 @@ class Relationship:
29
  class RelationshipEngine:
30
  """Engine for managing entity and event relationships."""
31
 
32
- def __init__(self, db_path: str = ':memory:') -> None:
33
  """Initialize the relationship engine with database connection."""
34
  self.conn = sqlite3.connect(db_path, check_same_thread=False)
 
35
  self.setup_database()
36
 
37
  def setup_database(self) -> None:
 
29
  class RelationshipEngine:
30
  """Engine for managing entity and event relationships."""
31
 
32
+ def __init__(self, db_path: str = ':memory:'):
33
  """Initialize the relationship engine with database connection."""
34
  self.conn = sqlite3.connect(db_path, check_same_thread=False)
35
+ self.conn.row_factory = sqlite3.Row # Critical fix
36
  self.setup_database()
37
 
38
  def setup_database(self) -> None:
ui/format.py CHANGED
@@ -17,11 +17,12 @@ class ResultFormatter:
17
 
18
  items = []
19
  for rel in relationships:
 
20
  item = f"""
21
  <li class="mb-2">
22
- <strong>{rel['source']}</strong> &rarr;
23
- <span class="text-blue-600">{rel['type'].replace('_to_', ' to ')}</span> &rarr;
24
- <strong>{rel['target']}</strong>
25
  <br/>
26
  <small class="text-gray-600">Confidence: {int(rel['confidence'] * 100)}%</small>
27
  </li>
 
17
 
18
  items = []
19
  for rel in relationships:
20
+ # Use correct SQL column names
21
  item = f"""
22
  <li class="mb-2">
23
+ <strong>{rel['source_text']}</strong> &rarr;
24
+ <span class="text-blue-600">{rel['relationship_type'].replace('_to_', ' to ')}</span> &rarr;
25
+ <strong>{rel['target_text']}</strong>
26
  <br/>
27
  <small class="text-gray-600">Confidence: {int(rel['confidence'] * 100)}%</small>
28
  </li>