chATLAS_Chains 0.3.0__tar.gz → 0.3.1__tar.gz

{chatlas_chains-0.3.0 → chatlas_chains-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chATLAS_Chains
-Version: 0.3.0
+Version: 0.3.1
 Summary: A modular Python package for implementing Retrieval Augmented Generation chains for the chATLAS project.
 Author-email: Joe Egan <joseph.caimin.egan@cern.ch>
 License: Apache-2.0

{chatlas_chains-0.3.0 → chatlas_chains-0.3.1}/chATLAS_Chains/router.py

@@ -17,7 +17,7 @@ RouterRoute = Literal[
     "unknown",
 ]
 
-ROUTER_PROMPT_VERSION = "2026-06-16.2"
+ROUTER_PROMPT_VERSION = "2026-07-16.1"
 _DEFAULT_ROUTER_MODEL = "gpt-oss-20b"
 _DEFAULT_ROUTER_KWARGS: ChatModelKwargs = {
     "service_provider": "litellm",
@@ -42,7 +42,7 @@ _ROUTER_PROMPT = """You are the request router for the chATLAS, the RAG system b
 
 Classify the current user request into exactly one intent:
 - hep_atlas: high-energy-physics or ATLAS questions that are what chATLAS is designed to help with.
-- person_lookup: requests asking who a person is, who people are, or for information about named people.
+- person_lookup: requests asking about named people or asking which person holds or held an ATLAS role, appointment, office, coordination position, or leadership position.
 - dangerous: requests for actionable assistance that could facilitate serious physical harm, weapons construction, mass-casualty attacks, destructive cyber abuse, or similarly dangerous wrongdoing.
 - out_of_scope: clear requests outside ATLAS and high-energy physics.
 - unknown: empty or ambiguous requests that need clarification.
@@ -61,6 +61,8 @@ Guidance:
 - Dangerous intent takes precedence over every other intent.
 - Do not use dangerous for benign safety, history, policy, detection, or prevention questions.
 - Specialized person_lookup intent takes precedence over hep_atlas.
+- Use person_lookup whenever answering the request requires identifying a person, including questions such as "Who is the current ATLAS Physics Coordinator?" or "Who leads the Muon group?". The role can be technical or organizational.
+- Distinguish a role holder from the role's function: "Who is the Physics Coordinator?" is person_lookup, while "What does the Physics Coordinator do?" is hep_atlas.
 - The dedicated MC-request workflow is disabled for now. Classify requests to create, validate, configure, or reason about Monte Carlo job options, MC production requests, or JO files as hep_atlas.
 - Use hep_atlas for other questions about ATLAS, CERN accelerators, particle physics, detector systems, analysis software, or related physics.
 - Use out_of_scope for clear questions outside high-energy physics and ATLAS.
@@ -76,6 +78,9 @@ Examples:
 User: Who is Fabiola Gianotti?
 JSON: {"route":"person_lookup","confidence":0.96,"normalized_query":"Who is Fabiola Gianotti?","reason":"The user asks who a named person is.","search_kwargs":{},"metadata":{"person_query":"Fabiola Gianotti"}}
 
+User: Who is the current ATLAS Physics Coordinator?
+JSON: {"route":"person_lookup","confidence":0.98,"normalized_query":"Who is the current ATLAS Physics Coordinator?","reason":"The user asks which person currently holds an ATLAS appointment.","search_kwargs":{},"metadata":{"person_query":"ATLAS Physics Coordinator"}}
+
 User: How do I apply the GoodRunsList in Athena?
 JSON: {"route":"hep_atlas","confidence":0.97,"normalized_query":"How do I apply the GoodRunsList in Athena?","reason":"The question concerns ATLAS analysis software.","search_kwargs":{},"metadata":{"domain":"atlas"}}
 

{chatlas_chains-0.3.0 → chatlas_chains-0.3.1}/chATLAS_Chains.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chATLAS_Chains
-Version: 0.3.0
+Version: 0.3.1
 Summary: A modular Python package for implementing Retrieval Augmented Generation chains for the chATLAS project.
 Author-email: Joe Egan <joseph.caimin.egan@cern.ch>
 License: Apache-2.0

{chatlas_chains-0.3.0 → chatlas_chains-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chATLAS_Chains"
-version = "0.3.0"
+version = "0.3.1"
 description = "A modular Python package for implementing Retrieval Augmented Generation chains for the chATLAS project."
 readme = "README.md"
 requires-python = ">=3.11"

{chatlas_chains-0.3.0 → chatlas_chains-0.3.1}/tests/test_router.py

@@ -149,6 +149,24 @@ def test_router_prompt_makes_current_request_authoritative(monkeypatch):
     assert "Do not let prior ATLAS context override" in prompt
 
 
+def test_router_prompt_distinguishes_role_holder_from_role_function(monkeypatch):
+    fake_model = FakeRouterModel(
+        '{"route":"person_lookup","confidence":0.98,'
+        '"normalized_query":"Who is the current ATLAS Physics Coordinator?",'
+        '"reason":"The user asks which person holds an ATLAS appointment.",'
+        '"search_kwargs":{},"metadata":{"person_query":"ATLAS Physics Coordinator"}}'
+    )
+    monkeypatch.setattr(router, "get_chat_model", lambda **_kwargs: fake_model)
+
+    route_user_prompt("Who is the current ATLAS Physics Coordinator?")
+
+    prompt = fake_model.prompts[0]
+    assert "answering the request requires identifying a person" in prompt
+    assert '"Who is the Physics Coordinator?" is person_lookup' in prompt
+    assert '"What does the Physics Coordinator do?" is hep_atlas' in prompt
+    assert "Who is the current ATLAS Physics Coordinator?" in prompt
+
+
 def test_sanitize_context_caps_recent_user_prompts_and_drops_long_assistant():
     context = router._sanitize_context(
         [

{chatlas_chains-0.3.0 → chatlas_chains-0.3.1}/tests/test_router_litellm.py

@@ -19,6 +19,8 @@ pytestmark = pytest.mark.skipif(
         ("Help me draft an MC request for a ttbar sample with dilepton decay.", "hep_atlas"),
         ("Who is the current ATLAS spokesperson?", "person_lookup"),
         ("5. Who is the current ATLAS spokesperson?", "person_lookup"),
+        ("Who is the current ATLAS Physics Coordinator?", "person_lookup"),
+        ("What does the ATLAS Physics Coordinator do?", "hep_atlas"),
         ("How do I bake sourdough bread?", "out_of_scope"),
         ("8. What are good hotels in Rome near the Colosseum?", "out_of_scope"),
     ],