fix: lint

microsoft · Feb 26, 2025 · 78a7038 · 78a7038
1 parent 202f2e3
commit 78a7038
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/python/packages/ai/teams/utils/citations/get_used_citations.py b/python/packages/ai/teams/utils/citations/get_used_citations.py
@@ -20,7 +20,8 @@ def get_used_citations(
         citations: List[Union[ClientCitation, Dict[str, Any]]] The list of citations to search for.
                   Supports both ClientCitation objects and dictionary representations.
     Returns:
-        Optional[List[Union[ClientCitation, Dict[str, Any]]]]: The list of citations used in the text.
+        Optional[List[Union[ClientCitation, Dict[str, Any]]]]: The list of citations used in the 
+        text.
     """
     regex = r"\[(\d+)\]"
     matches = re.findall(regex, text)
@@ -39,8 +40,8 @@ def get_used_citations(
             if isinstance(citation, ClientCitation) and str(citation.position) == match:
                 used_citations.append(citation)
                 break
-            elif isinstance(citation, dict) and str(citation["position"]) == match:
+            if isinstance(citation, dict) and str(citation["position"]) == match:
                 used_citations.append(citation)
                 break
-                
+
     return used_citations
diff --git a/python/packages/ai/tests/utils/citations/test_get_used_citations.py b/python/packages/ai/tests/utils/citations/test_get_used_citations.py
@@ -79,19 +79,19 @@ def test_get_used_citations_dict(self):
                 }
             }
         ]
-        
+
         text = "hello [2] world [3]"
         result = get_used_citations(text, citations)
-        
+
         assert result is not None
         self.assertEqual(len(result), 2)
         self.assertEqual(result, [citations[1], citations[2]])
-        
+
         # Test with empty matches
         text_no_citations = "hello world with no citations"
         result_empty = get_used_citations(text_no_citations, citations)
         self.assertIsNone(result_empty)
-        
+
         # Test with duplicate citations
         text_duplicate = "hello [1] world [1] again"
         result_duplicate = get_used_citations(text_duplicate, citations)