Skip to content

feat: expand individual entity type detection for diverse simulation scenarios #32

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
back2zion wants to merge 1 commit into
base: main
Choose a base branch
from
Open

feat: expand individual entity type detection for diverse simulation scenarios #32

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 14 additions & 3 deletions backend/app/services/oasis_profile_generator.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -168,7 +168,10 @@ class OasisProfileGenerator:
# Individual type entities (need to generate specific personas)
INDIVIDUAL_ENTITY_TYPES = [
"student", "alumni", "professor", "person", "publicfigure",
"expert", "faculty", "official", "journalist", "activist"
"expert", "faculty", "official", "journalist", "activist",
"consultant", "engineer", "architect", "researcher", "analyst",
"director", "manager", "leader", "cto", "ceo", "partner",
"developer", "scientist", "strategist", "advisor", "specialist"
]

# Group/institutional type entities (need to generate group representative personas)
Expand Down Expand Up @@ -431,8 +434,16 @@ def _build_entity_context(self, entity: EntityNode) -> str:
return "\n\n".join(context_parts)

def _is_individual_entity(self, entity_type: str) -> bool:
"""Determine if entity is an individual type"""
return entity_type.lower() in self.INDIVIDUAL_ENTITY_TYPES
"""Determine if entity is an individual type.

Falls back to treating unknown types as individual rather than group,
since the ontology generator often creates domain-specific types
(e.g. DataArchitect, AIConsultant) that aren't in either list.
"""
if entity_type.lower() in self.INDIVIDUAL_ENTITY_TYPES:
return True
# Unknown types default to individual rather than group
return entity_type.lower() not in self.GROUP_ENTITY_TYPES

def _is_group_entity(self, entity_type: str) -> bool:
"""Determine if entity is a group/institutional type"""
Expand Down