MCPcopy Create free account

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

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

↓ 4 callersFunctionenrich_anchor_documents
( analyzed_input: AnalyzedQuery | PartiallyAnalyzedQuery, inputs: PaperFinderInput, )
agents/mabool/api/mabool/agents/paper_finder/anchor_docs.py:9
↓ 4 callersFunctionfilter_docs_by_metadata
( docs: DocumentCollection, time_range: ExtractedYearlyTimeRange | None = None, venues: list[str]
agents/mabool/api/mabool/agents/common/common.py:115
↓ 4 callersFunctionget_llm_suggested_papers
( user_input: str, domains: DomainsIdentified, extra_hints: str | None = None, n_suggestions:
agents/mabool/api/mabool/agents/llm_suggestion/llm_suggestion_agent.py:86
↓ 4 callersFunctionget_publication_date_from_inserted_before
(inserted_before: str | None)
libs/dcollection/ai2i/dcollection/fetchers/s2.py:60
↓ 4 callersMethodinquiry
(self)
agents/mabool/api/mabool/infra/operatives/operatives.py:228
↓ 4 callersFunctionjoin_paragraphs
(*args: str)
agents/mabool/api/mabool/utils/text.py:8
↓ 4 callersFunctionkeyed_by_corpus_id
(documents: Sequence[Document])
libs/dcollection/ai2i/dcollection/collection.py:442
↓ 4 callersMethodload
(self, doc: Document, field: str)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:18
↓ 4 callersFunctionload_user_facing
(conf_dir: Path)
libs/config/ai2i/config/loading.py:69
↓ 4 callersMethodmanaged_scope
(self)
libs/di/ai2i/di/managed_env.py:61
↓ 4 callersMethodmap_n
( f: Callable[[OUT, OUT2], NEW_OUT], p1: ChainComputation[IN, OUT], p2: ChainComputati
libs/chain/ai2i/chain/computation.py:192
↓ 4 callersMethodmerge_dict
(self, d: dict[str, Any])
libs/config/ai2i/config/config_models.py:54
↓ 4 callersMethodopen_round_scope
(self, round_id: RoundId)
libs/di/ai2i/di/interface/app_context.py:41
↓ 4 callersMethodoverride
(self, updated_params: P)
libs/chain/ai2i/chain/models.py:492
↓ 4 callersMethodpredefined
(t: type[R], name: str, scope: Scope)
libs/di/ai2i/di/interface/models.py:79
↓ 4 callersFunctionremove_unicode_diacritics
(s: str)
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:113
↓ 4 callersMethodrerank
(self, rerank_input: RerankScoreInput)
agents/mabool/api/mabool/external_api/rerank/cohere.py:51
↓ 4 callersFunctionrerank_score_field
( relevance_criteria: RelevanceCriteria | None = None, field_name: str = "rerank_score" )
agents/mabool/api/mabool/agents/common/computed_fields/fields.py:25
↓ 4 callersMethodsearch
( self, content_query: str, relevance_criteria: RelevanceCriteria, domains: Do
agents/mabool/api/mabool/agents/broad_search_by_keyword/broad_search_by_keyword_agent.py:125
↓ 4 callersFunctionsigmoid
(x: float, center_shift: float = 0.0, steepness: float = 1.0)
agents/mabool/api/mabool/agents/common/sorting.py:397
↓ 4 callersMethodtimeout
(self, timeout: MaboolTimeout)
libs/chain/ai2i/chain/endpoints.py:242
↓ 3 callersFunction_aenter_with_error_handling
(mgr: AsyncContextManager[None], scope: ScopeContext)
libs/di/ai2i/di/scopes.py:273
↓ 3 callersFunction_choose_min_combined_span
Given a list of lists containing Snippet objects for each snippet part, this function selects one Snippet from each list such that the combin
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:151
↓ 3 callersFunction_get_config
()
libs/config/ai2i/config/config.py:37
↓ 3 callersFunction_get_user_facing
()
libs/config/ai2i/config/config.py:41
↓ 3 callersFunction_normalize
(x: Optional[str])
agents/mabool/api/mabool/data_model/specifications.py:36
↓ 3 callersFunctionadd_snowball_origins
( promoted_docs: DocumentCollection, variant: Literal["forward", "backward", "snippets"], search_iteration
agents/mabool/api/mabool/agents/snowball/snowball_utils.py:6
↓ 3 callersMethodapply
Compute a plan from the paper specification. :param paper_spec: The paper specification to resolve. :return: The resolved pla
agents/mabool/api/mabool/agents/metadata_only/plan.py:76
↓ 3 callersFunctionassign_to_origins
( docs_without_loaded_field: DocumentCollection, choose_best: bool = False )
agents/mabool/api/mabool/agents/dense/relevance_loading_optimization.py:150
↓ 3 callersMethodbuild
Build the operation with a specific factory.
agents/mabool/api/mabool/agents/metadata_only/ops.py:41
↓ 3 callersFunctioncheck_if_paper_inserted_before
( last_allowed_pub_date: date | None, year: int | None, publication_date: date | None )
libs/dcollection/ai2i/dcollection/fetchers/s2.py:72
↓ 3 callersMethodclone_with
(self, providers: Providers, pre_populated_cache: CachedValues | None = None)
libs/di/ai2i/di/managed_env.py:50
↓ 3 callersFunctioncounted_noun
(base: str, count: int)
agents/mabool/api/mabool/agents/common/step_reporting_utils.py:24
↓ 3 callersMethodcreate_fresh_scopes_context
(self, *, patched_instances: dict[str, Any])
libs/di/ai2i/di/app_context.py:50
↓ 3 callersMethoddestroy
(self, round_id: RoundId)
libs/di/ai2i/di/rounds_storage.py:50
↓ 3 callersFunctionexternal_dynamic_field
( *args: Any, fuse: Fuser[DocumentFieldName] = TakeFirst(), **kwargs: Any, )
libs/dcollection/ai2i/dcollection/interface/collection.py:88
↓ 3 callersMethodformat
(self, record: logging.LogRecord)
agents/mabool/api/mabool/utils/logging.py:29
↓ 3 callersMethodfrom_dense_retrieval
( queries: list[str], search_iteration: int, dataset: DenseDataset, top_k: int
agents/mabool/api/mabool/utils/dc.py:114
↓ 3 callersMethodfrom_s2_by_author
( authors_profiles: list[list[Any]], limit: int, dcf: DocumentCollectionFactory = DI.r
agents/mabool/api/mabool/utils/dc.py:52
↓ 3 callersMethodfrom_s2_by_title
( query: str, time_range: ExtractedYearlyTimeRange | None = None, venues: list[str] |
agents/mabool/api/mabool/utils/dc.py:64
↓ 3 callersMethodgemini25flash
Create Gemini 2.5 Flash reasoning model. Args: thinking_budget: Internal reasoning token budget (default: -1) - -
libs/chain/ai2i/chain/models.py:382
↓ 3 callersMethodgemini25pro
Create Gemini 2.5 Pro reasoning model. Args: thinking_budget: Internal reasoning token budget (default: -1) - -1:
libs/chain/ai2i/chain/models.py:406
↓ 3 callersMethodget_actual_vespa_version
(self)
libs/dcollection/ai2i/dcollection/external_api/dense/vespa.py:151
↓ 3 callersMethodget_model
(self, model_name: str)
libs/chain/ai2i/chain/tests/test_endpoints.py:125
↓ 3 callersMethodgpt41
(**params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:204
↓ 3 callersMethodgpt4o
(**params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:194
↓ 3 callersMethodgpt51
Create GPT-5.1 reasoning model with active reasoning. Args: reasoning_effort: Controls reasoning depth (default: "medium")
libs/chain/ai2i/chain/models.py:254
↓ 3 callersFunctiongreet
(name: str)
agents/mabool/api/mabool/services/tests/test_prioritized_task.py:150
↓ 3 callersMethodlookup_key
(self, entity_id: EntityId, field_name: DFN)
libs/dcollection/ai2i/dcollection/caching/cache.py:294
↓ 3 callersMethodnormalize
(self)
agents/mabool/api/mabool/data_model/specifications.py:187
↓ 3 callersMethodopen_scope
(self, values: CachedValues | None = None)
libs/di/ai2i/di/scopes.py:92
↓ 3 callersFunctionplan
(specifications: specs.Specifications)
agents/mabool/api/mabool/agents/metadata_only/plan.py:540
↓ 3 callersFunctionpluralize
(base: str, count: int)
agents/mabool/api/mabool/agents/common/step_reporting_utils.py:20
↓ 3 callersMethodpriority_context
(self, priority: int = DEFAULT_PRIORITY)
agents/mabool/api/mabool/services/prioritized_task.py:79
↓ 3 callersMethodproduct
(self, p: ChainComputation[IN, OUT2])
libs/chain/ai2i/chain/computation.py:176
↓ 3 callersMethodproject
(self, map_fn: DocumentProjector[V])
libs/dcollection/ai2i/dcollection/interface/collection.py:286
↓ 3 callersFunctionproject_root
()
agents/mabool/api/mabool/utils/paths.py:6
↓ 3 callersMethodput
Store entity field values in the cache.
libs/dcollection/ai2i/dcollection/caching/cache.py:206
↓ 3 callersMethodread_from_dict
(self, d: dict[str, Any])
libs/config/ai2i/config/config_models.py:78
↓ 3 callersMethodrun_paper_finder_on_analyzed_query
( self, inputs: PaperFinderInput, analyzed_input: AnalyzedQuery, anchor_docs: DocumentCollection )
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:247
↓ 3 callersFunctionscore_by_rank
(collection: DocumentCollection, max_score: int = 3)
agents/mabool/api/mabool/agents/specific_paper_by_name/specific_paper_by_name_agent.py:277
↓ 3 callersFunctionscoring_func
( extracted_name: str, found_in_search_with_content: int, title: str, abstract
agents/mabool/api/mabool/agents/specific_paper_by_name/specific_paper_by_name_agent.py:146
↓ 3 callersMethodsort
(self, docs: DocumentCollection)
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:484
↓ 3 callersMethodto_api_params
(self)
libs/chain/ai2i/chain/models.py:495
↓ 3 callersMethodto_flat_criteria
(self, include_nice_to_have: bool = True)
libs/dcollection/ai2i/dcollection/interface/document.py:369
↓ 3 callersFunctionto_reward
(doc: Document, field: str)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:140
↓ 3 callersMethodtotal_seconds
(self)
libs/chain/ai2i/chain/retry.py:247
↓ 3 callersFunctionverbalize_list
(items: list[str], connector: Literal["and", "or"])
agents/mabool/api/mabool/agents/common/step_reporting_utils.py:4
↓ 3 callersFunctionwith_priority
( func: Callable[P, Awaitable[T]], semaphore: PrioritySemaphore, priority: int = DEFAULT_PRIORITY )
agents/mabool/api/mabool/services/prioritized_task.py:100
↓ 3 callersMethodwith_trace_name
(self, name: str)
libs/chain/ai2i/chain/computation.py:91
↓ 2 callersMethod__call__
(self)
agents/mabool/api/mabool/agents/metadata_only/ops.py:65
↓ 2 callersMethod__init__
( self, scope: ScopeOrCustom, providers: Providers, patched_instances: dict[str, Any] | None = None
libs/di/ai2i/di/scopes.py:75
↓ 2 callersMethod_add_to_loaded_fields
(self, key: str)
libs/dcollection/ai2i/dcollection/document.py:136
↓ 2 callersFunction_count_relevance_by_level
(collection: DocumentCollection)
agents/mabool/api/mabool/agents/common/explain.py:125
↓ 2 callersFunction_create_and_wait_for_tasks
( session_id: str, operative_task: asyncio.Task[OperativeResponse[OUTPUT]], timeout: float = Confi
agents/mabool/api/mabool/infra/operatives/interaction.py:123
↓ 2 callersFunction_deep_lowercase_keys
(d: dict[str, Any])
libs/config/ai2i/config/loading.py:100
↓ 2 callersFunction_enrich_input_with
( input_type: type[PROMPT_PARAMS], f: Callable[[], Mapping[str, Any]] )
libs/chain/ai2i/chain/builders.py:279
↓ 2 callersFunction_extract_round_context
(request: Request, request_and_body: RequestAndBody)
agents/mabool/api/mabool/utils/context_deps.py:31
↓ 2 callersMethod_fetch_from_operatives
( self, operatives: list[SearchOperative], docs: DocumentCollection )
agents/mabool/api/mabool/agents/complex_search/broad_search.py:184
↓ 2 callersFunction_fetch_paper_data
( corpus_ids: Sequence[int], fields: Sequence[DocumentFieldName], s2_client: AsyncSemanticScholar,
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:126
↓ 2 callersMethod_follow_key
(self, d: dict[str, Any], key: str)
libs/config/ai2i/config/config_models.py:94
↓ 2 callersFunction_fuzzy_find_snippet_text
(text: str, snippet_to_find: str)
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:110
↓ 2 callersFunction_get_current_git_sha
()
agents/mabool/api/mabool/utils/file_based_cache.py:16
↓ 2 callersMethod_get_dynamic_field
(self, field_name: DocumentFieldName)
libs/dcollection/ai2i/dcollection/collection.py:166
↓ 2 callersMethod_get_first_completed_result
( self, done_tasks: set[asyncio.Task], pending_tasks: set[asyncio.Task] )
libs/chain/ai2i/chain/retry.py:199
↓ 2 callersFunction_get_qualified_name
(f: AsyncFunc[..., A] | AsyncContextFunc[..., A])
libs/di/ai2i/di/providers.py:215
↓ 2 callersMethod_is_set_type
Check if a node represents a set type.
dev/python/flake8_plugins/stable_sort_checker.py:32
↓ 2 callersMethod_merge_doc_source_responses
( self, docs: DocumentCollection, doc_source_responses: Sequence[OperativeResponse] )
agents/mabool/api/mabool/agents/complex_search/broad_search.py:229
↓ 2 callersMethod_register_dependency
(self, definition: DependencyDefinition[A])
libs/di/ai2i/di/providers.py:163
↓ 2 callersFunction_sub_task
()
libs/di/ai2i/di/tests/test_app_context.py:158
↓ 2 callersMethod_update_computed_fields
(self, fields: Sequence[DocumentFieldName | BaseComputedField])
libs/dcollection/ai2i/dcollection/collection.py:181
↓ 2 callersMethod_wait_for_completion
( self, active_tasks: list[asyncio.Task], timeout: float | None )
libs/chain/ai2i/chain/retry.py:194
↓ 2 callersMethod_wrap_value
(self, value: Any)
libs/config/ai2i/config/config_models.py:14
↓ 2 callersMethodabatch
( self, inputs: list[IN], config: RunnableConfig | list[RunnableConfig] | None = None,
libs/chain/ai2i/chain/retry.py:71
↓ 2 callersFunctionapi_key_mapper
()
libs/chain/ai2i/chain/tests/test_endpoints.py:179
↓ 2 callersMethodbuild
Build a cache keys lookup for the given entities and fields.
libs/dcollection/ai2i/dcollection/caching/cache.py:260
↓ 2 callersMethodcleanup
(self)
libs/di/ai2i/di/scopes.py:65
↓ 2 callersMethodclear_state
(self)
agents/mabool/api/mabool/infra/operatives/operatives.py:246
↓ 2 callersMethodclone_partial
(self, fields: list[DocumentFieldName] | None = None)
libs/dcollection/ai2i/dcollection/document.py:189
↓ 2 callersMethodcomputation
(self)
libs/dcollection/ai2i/dcollection/computed_field.py:49
← previousnext →101–200 of 1,349, ranked by callers