MCPcopy Create free account

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

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

⨍ Functions 1,349◇ Types & classes 423↳ Endpoints 5
↓ 1 callersMethod_extract_response_data
(self, response: Any)
libs/chain/ai2i/chain/gemini/async_genai.py:147
↓ 1 callersMethod_extract_text_from_response
(self, response: GenerateContentResponse, llm_output: dict)
libs/chain/ai2i/chain/gemini/async_genai.py:201
↓ 1 callersMethod_extract_usage_metadata
(self, response: GenerateContentResponse)
libs/chain/ai2i/chain/gemini/async_genai.py:172
↓ 1 callersMethod_fetch_from_doc_sources
( self, doc_sources: SearchIterationSources, docs: DocumentCollection )
agents/mabool/api/mabool/agents/complex_search/broad_search.py:214
↓ 1 callersMethod_fetch_missing_data
( self, query_fn: QueryFnSansContext[DFN], entities_to_query: set[DynamicallyLoadedEnt
libs/dcollection/ai2i/dcollection/caching/cache.py:138
↓ 1 callersFunction_fill_missing_corpus_ids
( docs: list[Document], docs_by_corpus_id: Mapping[CorpusId, Document], found_corpus_ids: Sequence
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:278
↓ 1 callersFunction_finalize_pending_tasks
( pending_tasks: set[asyncio.Task], session_id: str, is_timeout: bool, is_operative_done: bool, timeout: f
agents/mabool/api/mabool/infra/operatives/interaction.py:150
↓ 1 callersFunction_find_relevant_snippet_in_doc
(doc: Document, relevant_snippet: str)
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:22
↓ 1 callersFunction_find_snippet_parts
(text: str, relevant_snippet_parts: list[str])
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:106
↓ 1 callersFunction_flatten
(lst: list[Any])
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:261
↓ 1 callersMethod_format_messages
( self, messages: Sequence[BaseMessage] )
libs/chain/ai2i/chain/gemini/async_genai.py:109
↓ 1 callersFunction_get_app_state
(app: FastAPI)
libs/di/ai2i/di/integration.py:118
↓ 1 callersFunction_get_cited_corpus_ids_scores_df
( doc_collection: DocumentCollection, )
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:89
↓ 1 callersMethod_get_next_batch
(self)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:83
↓ 1 callersMethod_get_next_batch_size
(self)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:97
↓ 1 callersMethod_get_next_unloaded_document
(self, origin: Origin)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:106
↓ 1 callersFunction_handle_inquiry_reply
(query: InquiryReply, session_id: str)
agents/mabool/api/mabool/infra/operatives/interaction.py:116
↓ 1 callersMethod_handle_response
( response: OperativeResponse[BroadSearchOutput], )
agents/mabool/api/mabool/agents/complex_search/broad_search.py:304
↓ 1 callersMethod_has_next_page
(self)
libs/dcollection/ai2i/dcollection/external_api/s2/common.py:101
↓ 1 callersMethod_init_docs
(self, matches: dict)
libs/dcollection/ai2i/dcollection/external_api/dense/vespa.py:156
↓ 1 callersFunction_init_interactive_session
(allow_interactions: bool = True)
agents/mabool/api/mabool/infra/operatives/interaction.py:78
↓ 1 callersFunction_initial_retrieval
( content_query: str, domains: DomainsIdentified, authors: list[str] | None = None, venues: li
agents/mabool/api/mabool/agents/complex_search/fast_broad_search.py:130
↓ 1 callersMethod_initialize_iterators
(self)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:57
↓ 1 callersMethod_initialize_mab
(self)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:63
↓ 1 callersMethod_is_completed_task
(self, task: asyncio.Task)
libs/chain/ai2i/chain/retry.py:210
↓ 1 callersFunction_judge_documents_relevance
( documents: Sequence[Document], relevance_criteria: RelevanceCriteria, )
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:105
↓ 1 callersMethod_load_batch
(self, batch: OriginDocuments)
libs/dcollection/ai2i/dcollection/loaders/adaptive.py:114
↓ 1 callersFunction_load_relevance_judgement
( entities: Sequence[Document], relevance_criteria: RelevanceCriteria, metrics: Metrics = DI.requi
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:85
↓ 1 callersMethod_managed_async_func
(self, f: AsyncFunc[P, A])
libs/di/ai2i/di/resolver.py:172
↓ 1 callersMethod_managed_async_generator
(self, f: AsyncContextFunc[P, A])
libs/di/ai2i/di/resolver.py:192
↓ 1 callersMethod_managed_func
(self, f: Func[P, A])
libs/di/ai2i/di/resolver.py:182
↓ 1 callersMethod_managed_generator
(self, f: ContextFunc[P, A])
libs/di/ai2i/di/resolver.py:203
↓ 1 callersFunction_merge_duplicate_snippets
(snippets: list[Snippet])
libs/dcollection/ai2i/dcollection/fusers/fuse.py:39
↓ 1 callersFunction_prepare_documents_for_llm
( documents: Sequence[Document], criteria: list[RelevanceCriterion] )
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:124
↓ 1 callersMethod_process_cache_results
( self, entities: Sequence[DynamicallyLoadedEntity[DFN]], fields: Sequence[FieldRequir
libs/dcollection/ai2i/dcollection/caching/cache.py:72
↓ 1 callersFunction_process_llm_results_with_failures
( llm_results: list[LLMJudgementResult | Exception], original_documents: Sequence[Document], relev
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:205
↓ 1 callersMethod_process_model_kwargs
(self, kwargs: dict[str, Any], stop: list[str] | None = None)
libs/chain/ai2i/chain/gemini/async_genai.py:82
↓ 1 callersMethod_race_calls
(self, input: IN, config: RunnableConfig | None, soft_timeout: float, **kwargs: Any)
libs/chain/ai2i/chain/retry.py:137
↓ 1 callersMethod_race_with_fallbacks
( self, original_task: asyncio.Task, input: IN, config: RunnableConfig | None, soft_timeout: float, **
libs/chain/ai2i/chain/retry.py:159
↓ 1 callersMethod_refresh_iteration_shortcircuit
(self)
agents/mabool/api/mabool/agents/complex_search/broad_search.py:179
↓ 1 callersFunction_relevance_judgement_score
(relevance_judgement: RelevanceJudgement | None)
agents/mabool/api/mabool/agents/common/sorting.py:512
↓ 1 callersMethod_relevance_score_for_cap
(cls, doc: Document)
agents/mabool/api/mabool/agents/dense/relevance_loading_optimization.py:79
↓ 1 callersFunction_render_relevant_snippet_text
(part: str)
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:75
↓ 1 callersFunction_rerank_docs
( doc_collection: DocumentCollection, relevance_criteria: RelevanceCriteria | None = None, cohere_rerank:
agents/mabool/api/mabool/agents/complex_search/fast_broad_search.py:142
↓ 1 callersFunction_run_initial_retrieval
( content_query: str, domains: DomainsIdentified, authors: list[str] | None = None, venues: li
agents/mabool/api/mabool/agents/complex_search/fast_broad_search.py:63
↓ 1 callersFunction_run_operative
( operative: Operative, session_id: str, query: Any, )
agents/mabool/api/mabool/infra/operatives/interaction.py:102
↓ 1 callersFunction_run_relevance_judgement
( doc_collection: DocumentCollection, relevance_criteria: RelevanceCriteria )
agents/mabool/api/mabool/agents/complex_search/fast_broad_search.py:195
↓ 1 callersFunction_s2_get_paper_data_for_missing_id
( missing_corpus_id: CorpusId, fields: Sequence[DocumentFieldName], s2_client: AsyncSemanticSchola
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:301
↓ 1 callersFunction_sigmoid
(x: float)
agents/mabool/api/mabool/agents/common/sorting.py:393
↓ 1 callersFunction_sorted_by_origin_ranks
(doc_collection: DocumentCollection)
agents/mabool/api/mabool/agents/complex_search/fast_broad_search.py:165
↓ 1 callersFunction_titles_match
(title1: str | None, title2: str | None)
agents/mabool/api/mabool/agents/specific_paper_by_title/specific_paper_by_title_agent.py:62
↓ 1 callersFunction_to_docs
( docs_by_corpus_id: Mapping[CorpusId, Document], fields: Sequence[DocumentFieldName], papers: Seq
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:198
↓ 1 callersFunction_to_original_entities_order
( entities: Sequence[Document], judgements_by_id: dict[CorpusId, RelevanceJudgement], failed_doc_e
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:274
↓ 1 callersFunction_try_create_document_judgement
( doc: Document, scores: RelevanceScores, judgement_data: list[dict[str, Any]], relevance_crit
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:294
↓ 1 callersMethod_try_original_with_timeout
( self, input: IN, config: RunnableConfig | None, soft_timeout: float, **kwargs: Any )
libs/chain/ai2i/chain/retry.py:148
↓ 1 callersMethod_update_computed_field
(self, computed_field: BaseComputedField)
libs/dcollection/ai2i/dcollection/collection.py:189
↓ 1 callersMethod_validate_args
(self)
agents/mabool/api/mabool/agents/complex_search/broad_search.py:171
↓ 1 callersFunction_validate_required_criteria
Validate that all required criteria are present in judgements.
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:325
↓ 1 callersMethod_validate_response
(response: Response)
libs/dcollection/ai2i/dcollection/external_api/dense/vespa.py:206
↓ 1 callersMethod_wait_for_remaining_tasks
(self, active_tasks: list[asyncio.Task], fallback_task: asyncio.Task)
libs/chain/ai2i/chain/retry.py:183
↓ 1 callersMethodaccumulate
(self, docs: list[Document])
agents/mabool/api/mabool/agents/dense/relevance_loading_optimization.py:39
↓ 1 callersFunctionadaptive_load
( documents: DocumentCollection, field_name: DocumentFieldName = "relevance_judgement", docs_quota
agents/mabool/api/mabool/agents/dense/relevance_loading_optimization.py:89
↓ 1 callersFunctionadd_broad_search_score
( collection: DocumentCollection, sort_documents_input: SortPreferences, only_use_existing_fields:
agents/mabool/api/mabool/agents/common/sorting.py:520
↓ 1 callersMethodadd_relevance_judged
(self, corpus_id: str)
agents/mabool/api/mabool/utils/metrics.py:15
↓ 1 callersFunctionadd_weighted_sort_score
( collection: DocumentCollection, sort_documents_input: SortPreferences, )
agents/mabool/api/mabool/agents/common/sorting.py:341
↓ 1 callersFunctionanalyze
( user_input: str, fields: ExtractedFields, specifications: Specifications, endpoint: LLMEndpoint )
agents/mabool/api/mabool/agents/query_analyzer/query_analyzer.py:118
↓ 1 callersFunctionask_for_option
(inquiry: Inquiry, attempts: int)
agents/mabool/api/mabool/agents/query_refusal/utils.py:80
↓ 1 callersMethodassert_all_called
(self, model_names: list[str])
libs/chain/ai2i/chain/tests/test_endpoints.py:142
↓ 1 callersMethodassign_loaded_values
( self, field_names: Sequence[DocumentFieldName], loaded_entities: Sequence[Document],
libs/dcollection/ai2i/dcollection/document.py:217
↓ 1 callersFunctionasync_func
( value: str = DI.requires(my_value), )
libs/di/ai2i/di/tests/test_managed_env.py:378
↓ 1 callersFunctionasync_gen
( value: str = DI.requires(my_value), )
libs/di/ai2i/di/tests/test_managed_env.py:384
↓ 1 callersMethodattempt_metadata_only_search
(self)
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:472
↓ 1 callersFunctionbreak_iteration_context
()
agents/mabool/api/mabool/agents/complex_search/broad_search.py:340
↓ 1 callersFunctionbuild_filters
( time_range: ExtractedYearlyTimeRange | None = None, venues: list[str] | None = None, authors: li
libs/dcollection/ai2i/dcollection/external_api/dense/vespa.py:48
↓ 1 callersMethodbuild_from_name
(self, name: DependencyName, resolver: DependencyResolver)
libs/di/ai2i/di/providers.py:68
↓ 1 callersMethodcache
(self)
libs/dcollection/ai2i/dcollection/factory.py:72
↓ 1 callersMethodcache
(self)
libs/dcollection/ai2i/dcollection/interface/collection.py:177
↓ 1 callersFunctioncapitalized_weirdly
(name: str)
agents/mabool/api/mabool/agents/specific_paper_by_name/specific_paper_by_name_agent.py:96
↓ 1 callersFunctioncentral_first_score_sigmoid
( citation_count: int | None, center_citation_count: int = 50, steepness: float = 1.8, )
agents/mabool/api/mabool/agents/common/sorting.py:401
↓ 1 callersMethodchain_with
(self, other: ProvidersPerScope)
libs/di/ai2i/di/scopes.py:411
↓ 1 callersFunctionclarify_query_refusal
( inquiry: Inquiry | None, analyzed_input: AnalyzedQuery | PartiallyAnalyzedQuery )
agents/mabool/api/mabool/agents/query_refusal/utils.py:77
↓ 1 callersFunctionclean_overlaps
(sorted_snippets: list[Snippet])
libs/dcollection/ai2i/dcollection/fusers/fuse.py:52
↓ 1 callersMethodclear
Clear all cached data.
libs/dcollection/ai2i/dcollection/caching/cache.py:229
↓ 1 callersMethodclear_loaded_field
(self, field_name: DocumentFieldName)
libs/dcollection/ai2i/dcollection/document.py:146
↓ 1 callersMethodclose_round_scope
(self)
libs/di/ai2i/di/app_context.py:113
↓ 1 callersFunctioncollect_values
Recursively collect all string values from nested dictionaries.
libs/config/ai2i/config/gen_config_types.py:139
↓ 1 callersFunctioncombine_content_query_with_anchors
(content_query: str, anchor_docs: DocumentCollection)
agents/mabool/api/mabool/agents/query_analyzer/anchor.py:46
↓ 1 callersMethodconsumes
Abstract method to define the consumed attributes.
agents/mabool/api/mabool/agents/metadata_only/plan.py:41
↓ 1 callersFunctioncontent_relevance_score
( relevance_judgement_score: float, rerank_score: float, num_snippets: int, original_order_score: float =
agents/mabool/api/mabool/agents/common/sorting.py:482
↓ 1 callersMethodcontinue_round_scope
(self)
libs/di/ai2i/di/app_context.py:110
↓ 1 callersFunctionconvert_choice_to_valid_number
(user_choice: str)
agents/mabool/api/mabool/agents/query_refusal/utils.py:101
↓ 1 callersFunctionconvert_gemini25_flash_thinking_params
Convert Gemini 2.5 thinking parameters to Google GenAI format. Key behavior: - thinking_budget controls internal reasoning token budget -
libs/chain/ai2i/chain/params.py:170
↓ 1 callersFunctioncreate_app
(di_patched_instances: dict[str, Any] | None = None, **config_overrides: dict[str, Any])
agents/mabool/api/mabool/api/app.py:25
↓ 1 callersFunctioncreate_empty_app_context
()
libs/di/ai2i/di/factory/app_context.py:10
↓ 1 callersFunctioncreate_json_response
(content: Any)
agents/mabool/api/mabool/api/route_utils.py:6
↓ 1 callersFunctioncreate_mock_app
(**config_overrides: dict[str, Any])
agents/mabool/api/mabool/infra/operatives/tests/mock_app.py:125
↓ 1 callersMethodcreate_scopes
(self, *, patched_instances: dict[str, Any])
libs/di/ai2i/di/scopes.py:419
↓ 1 callersFunctiondecompose_and_analyze_query
( user_input: str, )
agents/mabool/api/mabool/agents/query_analyzer/query_analyzer.py:271
↓ 1 callersFunctiondecompose_and_analyze_query_restricted
( user_input: str, )
agents/mabool/api/mabool/agents/query_analyzer/query_analyzer.py:287
↓ 1 callersFunctiondeep_merge
Recursively merge two dictionaries.
libs/config/ai2i/config/gen_config_types.py:127
← previousnext →301–400 of 1,349, ranked by callers