MCPcopy Create free account

hub / github.com/allenai/asta-paper-finder / functions

Functions1,349 in github.com/allenai/asta-paper-finder

↓ 2 callersMethodcontinue_scope
(self)
libs/di/ai2i/di/scopes.py:170
↓ 2 callersFunctioncreate_app_context
(module: Module)
libs/di/ai2i/di/factory/app_context.py:6
↓ 2 callersFunctioncreate_managed_app
( app_context: ApplicationContext, app_config: ConfigSettings, di_patched_instances: dict[str, Any], conf_
libs/di/ai2i/di/integration.py:17
↓ 2 callersFunctioncustom_async_client
(**config_overrides: Any)
agents/mabool/api/mabool/infra/operatives/tests/test_interaction.py:21
↓ 2 callersMethoddestroy_round_scope
(self, round_id: RoundId)
libs/di/ai2i/di/interface/app_context.py:53
↓ 2 callersFunctionexplain_query_analysis
(analyzed_input: AnalyzedQuery | PartiallyAnalyzedQuery)
agents/mabool/api/mabool/agents/common/explain.py:63
↓ 2 callersFunctionfilter_by_author
(expected_authors: list[str], doc: Document, keep_missing: bool | None = False)
agents/mabool/api/mabool/agents/common/common.py:85
↓ 2 callersMethodfrom_dict
(cls, journal: S2Journal)
libs/dcollection/ai2i/dcollection/interface/document.py:81
↓ 2 callersMethodfrom_s2_citing_papers
( corpus_id: CorpusId, search_iteration: int = 1, dcf: DocumentCollectionFactory = DI.
agents/mabool/api/mabool/utils/dc.py:102
↓ 2 callersMethodfuse
(self, *entities: Document)
libs/dcollection/ai2i/dcollection/document.py:178
↓ 2 callersMethodgemini2flash
(**params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:244
↓ 2 callersMethodgemini3pro
Create Gemini 3 Pro reasoning model. Args: thinking_level: Reasoning depth control (default: "high") - "low": Min
libs/chain/ai2i/chain/models.py:441
↓ 2 callersFunctionget_application_config
(app_state: AppState = Depends(_get_app_state))
libs/di/ai2i/di/integration.py:122
↓ 2 callersFunctionget_by_title_origin_query
( query: str, time_range: ExtractedYearlyTimeRange | None = None, venues: Sequence[str] | None = N
libs/dcollection/ai2i/dcollection/fetchers/s2.py:85
↓ 2 callersFunctionget_config_or_throw
()
libs/config/ai2i/config/config.py:52
↓ 2 callersFunctionget_default_endpoint
()
agents/mabool/api/mabool/agents/query_analyzer/query_analyzer.py:48
↓ 2 callersFunctionget_default_endpoint
()
agents/mabool/api/mabool/agents/llm_suggestion/llm_suggestion_agent.py:49
↓ 2 callersFunctionget_dense_datasets_by_domains
(domains: DomainsIdentified)
agents/mabool/api/mabool/agents/common/domain_utils.py:26
↓ 2 callersMethodget_predefined_dynamic_fields
()
libs/dcollection/ai2i/dcollection/document.py:41
↓ 2 callersFunctionget_singleton_env
(app_state: AppState = Depends(_get_app_state))
libs/di/ai2i/di/integration.py:126
↓ 2 callersFunctionget_specific_paper_by_name
( user_input: str, extracted_name: str, domains: DomainsIdentified, extracted_content: str | N
agents/mabool/api/mabool/agents/specific_paper_by_name/specific_paper_by_name_agent.py:437
↓ 2 callersFunctionget_system_domain_params
(domains: DomainsIdentified)
agents/mabool/api/mabool/agents/common/domain_utils.py:11
↓ 2 callersMethodgpt4turbo
(**params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:224
↓ 2 callersMethodgpt51_no_reasoning
Create GPT-5.1 with reasoning disabled (reasoning_effort='none'). Args: temperature: Sampling temperature (configurable when reas
libs/chain/ai2i/chain/models.py:307
↓ 2 callersMethodgroup_by
(self, group_fn: DocumentProjector[V])
libs/dcollection/ai2i/dcollection/interface/collection.py:289
↓ 2 callersMethodinit_root_manager
(id: StateManagerId)
agents/mabool/api/mabool/infra/operatives/operatives.py:171
↓ 2 callersFunctionis_timeout_error
(error: BaseException | None)
libs/chain/ai2i/chain/retry.py:214
↓ 2 callersFunctionlog_relevance_value_counts
( documents: DocumentCollection, )
agents/mabool/api/mabool/agents/common/relevance_judgement_utils.py:48
↓ 2 callersMethodmanaged
(self, f: AsyncFunc[P, A])
libs/di/ai2i/di/resolver.py:145
↓ 2 callersMethodmany
( self, inputs: list[IN], *, return_exceptions: TrueLiteral, **batch_ec: Unpack[BatchExecutionContextB
libs/chain/ai2i/chain/endpoints.py:190
↓ 2 callersMethodmap_enumerate
(self, map_fn: DocumentEnumProjector[Document])
libs/dcollection/ai2i/dcollection/interface/collection.py:277
↓ 2 callersFunctionmerge_time_range_and_inserted_before
( time_range: ExtractedYearlyTimeRange | None, inserted_before: str | None )
libs/dcollection/ai2i/dcollection/fetchers/s2.py:37
↓ 2 callersMethodmodel_factory
( structured_response: bool, model: LLMModel[Any], timeout: MaboolTimeout,
libs/chain/ai2i/chain/tests/test_endpoints.py:129
↓ 2 callersFunctionnormalize_column
(s: pd.Series[float], s_min: float | None = None, s_max: float | None = None)
agents/mabool/api/mabool/agents/common/sorting.py:153
↓ 2 callersFunctionnormalize_title
(t: str)
agents/mabool/api/mabool/agents/specific_paper_by_title/specific_paper_by_title_agent.py:63
↓ 2 callersFunctionplot_relevance_distribution
(documents: DocumentCollection, strategy: str, timestamp: str | None = None)
agents/mabool/api/mabool/agents/dense/relevance_loading_optimization.py:241
↓ 2 callersFunctionrefusal_responses
()
agents/mabool/api/mabool/agents/query_refusal/utils.py:38
↓ 2 callersMethodreopen_round_scope
(self, round_id: RoundId)
libs/di/ai2i/di/interface/app_context.py:44
↓ 2 callersMethodreplace_env
(self, env: ManagedEnv, cleanup: CleanUpFunc | None = None)
libs/di/ai2i/di/scopes.py:52
↓ 2 callersFunctionreport_relevance_judgement_counts
(documents: DocumentCollection)
agents/mabool/api/mabool/agents/common/relevance_judgement_utils.py:41
↓ 2 callersMethodretry_specific_in_broad_mode
( self, inputs: PaperFinderInput, analyzed_query: AnalyzedQuery, anchor_docs: DocumentCollection )
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:223
↓ 2 callersMethodrun
Run the operation and return a result.
agents/mabool/api/mabool/agents/metadata_only/ops.py:61
↓ 2 callersMethodrun_paper_finder_on_partially_analyzed_query
( self, inputs: PaperFinderInput, analyzed_input: PartiallyAnalyzedQuery, errors: list[QueryAnalyzerEr
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:327
↓ 2 callersFunctionrun_snippet_snowball
( content_query: str, doc_collection: DocumentCollection, top_k: int, search_iteration: int,
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:349
↓ 2 callersMethods2_client
(self)
libs/dcollection/ai2i/dcollection/factory.py:69
↓ 2 callersFunctionscore_snowball_candidate
( citations: pd.DataFrame, direction: Literal["forward", "backward"], seed_relevance_bias: float,
agents/mabool/api/mabool/agents/snowball/snowball_agent.py:338
↓ 2 callersMethodshould_break
(self)
agents/mabool/api/mabool/agents/dense/relevance_loading_optimization.py:43
↓ 2 callersFunctionsuggest_retrieval_query
(paper_description: str)
agents/mabool/api/mabool/agents/broad_search_by_keyword/broad_search_by_keyword_agent.py:42
↓ 2 callersMethodsuspend_runnable
(builder: Callable[[], Runnable[A, B]])
libs/chain/ai2i/chain/computation.py:85
↓ 2 callersFunctiontask1
()
agents/mabool/api/mabool/services/tests/test_prioritized_task.py:78
↓ 2 callersFunctiontask2
()
agents/mabool/api/mabool/services/tests/test_prioritized_task.py:83
↓ 2 callersFunctiontask3
()
agents/mabool/api/mabool/services/tests/test_prioritized_task.py:88
↓ 2 callersMethodupdate_computed_fields
(self, fields: Sequence[DocumentFieldName | BaseComputedField])
libs/dcollection/ai2i/dcollection/interface/collection.py:303
↓ 2 callersFunctionweighted_sort_calculation
( documents: Sequence[Document], weights: Weights | dict[str, float], )
agents/mabool/api/mabool/agents/common/sorting.py:307
↓ 2 callersMethodwith_default
(self, v: B)
libs/config/ai2i/config/config_models.py:88
↓ 2 callersMethodwith_default_factory
(self, f: Callable[[], B])
libs/config/ai2i/config/config_models.py:91
↓ 1 callersMethod__anext__
(self)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:77
↓ 1 callersMethod__init__
( self, decorated: Runnable[IN, OUT], retry_settings: TenacityRetrySettings, should_retry_on_timeout:
libs/chain/ai2i/chain/retry.py:41
↓ 1 callersMethod__init__
(self)
libs/chain/ai2i/chain/tests/test_endpoints.py:107
↓ 1 callersMethod__init__
(self, message: Optional[str] = None)
agents/mabool/api/mabool/data_model/agent.py:219
↓ 1 callersFunction_add_missing_fields_and_sort
( initial_docs: DocumentCollection, plan_context: PlanContext )
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:92
↓ 1 callersMethod_agenerate_message
( self, messages: Sequence[BaseMessage], stop: list[str] | None = None, **kwargs: Any )
libs/chain/ai2i/chain/gemini/async_genai.py:62
↓ 1 callersFunction_align_ref_mentions
( snippet: Snippet, sentences: list[Snippet], sentence_offsets: list[SentenceOffsets] )
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:185
↓ 1 callersMethod_apply_all_values
( self, entities: Sequence[DynamicallyLoadedEntity[DFN]], fields: set[DFN], ca
libs/dcollection/ai2i/dcollection/caching/cache.py:114
↓ 1 callersMethod_apply_relevance_judgement
(self, docs: DocumentCollection)
agents/mabool/api/mabool/agents/complex_search/broad_search.py:240
↓ 1 callersFunction_async_context_adapter
( context_factory: AsyncContextFunc[..., A], provider_name: str )
libs/di/ai2i/di/providers.py:189
↓ 1 callersFunction_async_factory_adapter
( factory: AsyncFactory[A], provider_name: str )
libs/di/ai2i/di/providers.py:174
↓ 1 callersFunction_batched_from_s2
(entities: Sequence[Document], fields: Sequence[DocumentFieldName])
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:55
↓ 1 callersMethod_build_all
(self)
libs/di/ai2i/di/managed_env.py:68
↓ 1 callersFunction_build_merged_response_potentially_relevant
(prev_summary: str, relevant_docs: Sequence[Document])
agents/mabool/api/mabool/agents/common/explain.py:243
↓ 1 callersFunction_build_merged_response_with_perfect_matches
( prev_summary: str, perfect_count: int, high_count: int, other_count: int )
agents/mabool/api/mabool/agents/common/explain.py:250
↓ 1 callersFunction_build_merged_response_without_perfect_matches
(prev_summary: str, high_count: int, other_count: int)
agents/mabool/api/mabool/agents/common/explain.py:274
↓ 1 callersFunction_build_relevance_criteria_judgements
( doc: Document, judgement_data: list[dict[str, Any]] )
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:392
↓ 1 callersFunction_build_relevance_judgement
Build the final RelevanceJudgement object.
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:365
↓ 1 callersFunction_build_response_with_perfect_matches
(perfect_count: int, high_count: int, other_count: int)
agents/mabool/api/mabool/agents/common/explain.py:162
↓ 1 callersFunction_build_response_without_perfect_matches
(high_count: int, other_count: int)
agents/mabool/api/mabool/agents/common/explain.py:183
↓ 1 callersMethod_cache_assigned_values
( self, sorted_entities: list[DynamicallyLoadedEntity[DFN]], fields_to_query: set[DFN]
libs/dcollection/ai2i/dcollection/caching/cache.py:166
↓ 1 callersMethod_cache_fetched_data
( self, fetched_entities: list[DynamicallyLoadedEntity[DFN]], fields_to_query: set[DFN
libs/dcollection/ai2i/dcollection/caching/cache.py:152
↓ 1 callersMethod_cache_none_values
( self, sorted_entities: list[DynamicallyLoadedEntity[DFN]], fields_to_query: set[DFN]
libs/dcollection/ai2i/dcollection/caching/cache.py:185
↓ 1 callersFunction_calculate_relevance_criteria_score
Calculate weighted relevance score based on criteria weights (RESTORED from original).
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:417
↓ 1 callersMethod_check_pandas_sort
Check if pandas sort method has kind='stable'.
dev/python/flake8_plugins/stable_sort_checker.py:25
↓ 1 callersFunction_cited_corpus_ids_to_df
(cited_corpus_ids: dict[str, Any])
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:60
↓ 1 callersMethod_cleanup
()
libs/di/ai2i/di/scopes.py:103
↓ 1 callersMethod_cleanup
()
libs/di/ai2i/di/scopes.py:163
↓ 1 callersFunction_collect_cited_corpus_ids
(doc_collection: DocumentCollection)
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:27
↓ 1 callersMethod_collect_doc_sources
( self, shortcircuit: Shortcircuit, dense_agent: DenseAgent, snowball_agent: S
agents/mabool/api/mabool/agents/complex_search/broad_search.py:132
↓ 1 callersFunction_compute_document_relevance_score
( doc_id: CorpusId, criterion_judgements: list[RelevanceCriterionJudgement], relevance_criteria: R
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:351
↓ 1 callersFunction_convert_relevance_score_to_level
Convert relevance score to 0-3 level (UPDATED to use new threshold constants).
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:433
↓ 1 callersFunction_convert_to_relevance_scores
(results: list[LLMJudgementResult], metrics: Metrics)
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:243
↓ 1 callersMethod_create_batches
(self)
libs/common/ai2i/common/utils/batch.py:64
↓ 1 callersFunction_create_dynamic_judgement_result_type
(criteria: list[RelevanceCriterion])
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:159
↓ 1 callersFunction_create_relevance_judgement_chain
(criteria: list[RelevanceCriterion])
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:139
↓ 1 callersFunction_create_retry_logger
()
libs/chain/ai2i/chain/builders.py:298
↓ 1 callersFunction_default_batch_execution_context
()
libs/chain/ai2i/chain/endpoints.py:121
↓ 1 callersMethod_determine_role
(self, message: BaseMessage)
libs/chain/ai2i/chain/gemini/async_genai.py:139
↓ 1 callersFunction_doc_best_rank
(doc: Document)
agents/mabool/api/mabool/agents/complex_search/fast_broad_search.py:167
↓ 1 callersFunction_document_fields_to_s2_fields
( fields: Sequence[DocumentFieldName], )
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:78
↓ 1 callersFunction_enrich_anchor_document
(query: str, corpus_id: CorpusId)
agents/mabool/api/mabool/agents/paper_finder/anchor_docs.py:27
↓ 1 callersFunction_execute_llm_judgements
( inputs: list[DocumentRelevanceInput], judge_relevance_chain: ChainComputation )
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:185
↓ 1 callersMethod_extract_llm_metadata
(self, response: GenerateContentResponse)
libs/chain/ai2i/chain/gemini/async_genai.py:157
← previousnext →201–300 of 1,349, ranked by callers