MCPcopy Create free account

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

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

↓ 94 callersFunctionufv
(reader: ConfigValuePlaceholder[str], /, **format_args: Any)
libs/config/ai2i/config/config.py:177
↓ 74 callersFunctionconfig_value
( reader: ConfigValuePlaceholder[A], /, default: B | ValueNotSet = ValueNotSet.instance(), def
libs/config/ai2i/config/config.py:155
↓ 63 callersMethodget_dependency
( self, definition: DependencyDefinition[A], /, default: B | ValueNotSet = Val
libs/di/ai2i/di/resolver.py:87
↓ 58 callersMethodrequires
( self, definition: DependencyDefinition[A], /, default: B | ValueNotSet = Val
libs/di/ai2i/di/interface/gateway.py:42
↓ 48 callersMethodwith_fields
( self, fields: Sequence[DocumentFieldName | BaseComputedField[DocumentFieldName, Any]] )
libs/dcollection/ai2i/dcollection/interface/collection.py:298
↓ 39 callersMethodmap
(self, g: Callable[[OUT], NEW_OUT])
libs/chain/ai2i/chain/computation.py:104
↓ 28 callersFunctiondefine_prompt_llm_call
( template: str, /, input_type: type[PROMPT_PARAMS], output_type: type[LLM_STRUCTURED_OUTPUT],
libs/chain/ai2i/chain/builders.py:50
↓ 28 callersMethodempty
(dcf: DocumentCollectionFactory = DI.requires(dc_deps.round_doc_collection_factory))
agents/mabool/api/mabool/utils/dc.py:39
↓ 25 callersMethodbuild_runnable
(self, mf: ModelRunnableFactory)
libs/chain/ai2i/chain/computation.py:88
↓ 23 callersMethodmanaged_scope
(self, values: CachedValues | None = None)
libs/di/ai2i/di/interface/scopes.py:45
↓ 22 callersMethodget_dependency_by_name
( self, name: str, /, default: Any | ValueNotSet = ValueNotSet.instance(),
libs/di/ai2i/di/resolver.py:77
↓ 22 callersMethodonce
(self, input: IN)
libs/chain/ai2i/chain/endpoints.py:184
↓ 21 callersFunctiondefine_model
(*, structured_response: bool)
libs/chain/ai2i/chain/builders.py:288
↓ 21 callersFunctiondynamic_field
( *args: Any, loaders: Sequence[DocumentFieldLoader] | None = None, fuse: Fuser[DocumentFieldName]
libs/dcollection/ai2i/dcollection/interface/collection.py:78
↓ 20 callersFunctionConfigValue
( # noqa: N802 reader: ConfigValuePlaceholder[A], /, default: B | ValueNotSet = ValueNotSet.insta
libs/config/ai2i/config/config.py:141
↓ 20 callersMethodainvoke
(self, input: PromptValue, config: RunnableConfig | None = None, **kwargs: Any)
libs/chain/ai2i/chain/tests/mocks.py:19
↓ 20 callersMethodfilter
Filter a plan from the paper specification. :param paper_spec: The paper specification to resolve. :param prev: The previous
agents/mabool/api/mabool/agents/metadata_only/plan.py:98
↓ 20 callersMethodlift
(runnable: Runnable[A, B])
libs/chain/ai2i/chain/computation.py:69
↓ 18 callersMethodinstance
()
libs/common/ai2i/common/utils/value.py:18
↓ 16 callersFunction_create_dep
(name: str, deps: list[str] | None = None)
libs/di/ai2i/di/tests/test_utils.py:89
↓ 16 callersFunctiondefine_chat_llm_call
( messages: Sequence[ChatMessage], /, input_type: type[PROMPT_PARAMS], output_type: type[LLM_S
libs/chain/ai2i/chain/builders.py:188
↓ 16 callersFunctionsystem_message
(text: str)
libs/chain/ai2i/chain/builders.py:172
↓ 16 callersFunctionuser_message
(text: str)
libs/chain/ai2i/chain/builders.py:176
↓ 15 callersMethodfetch_async_data
( self, query_fn: QueryFnSansContext[DFN], entities: Sequence[DynamicallyLoadedEntity[
libs/dcollection/ai2i/dcollection/caching/cache.py:35
↓ 15 callersMethodfrom_name
(name: ModelName, **params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:79
↓ 15 callersFunctionget_value
(k: int = ConfigValue(cfg_schema.dense_agent.initial_top_k_per_query))
libs/config/ai2i/config/tests/test_config.py:37
↓ 15 callersMethodinit_operative
( self, id: str, operative_type: Type[OPERATIVE], interactions: Interactions |
agents/mabool/api/mabool/infra/operatives/operatives.py:234
↓ 15 callersMethodpop
(self, round_id: RoundId)
libs/di/ai2i/di/rounds_storage.py:42
↓ 15 callersFunctionto_field_reqs
(fields: list[str])
libs/dcollection/ai2i/dcollection/caching/tests/test_cache.py:75
↓ 14 callersMethodexecute
(self)
agents/mabool/api/mabool/agents/complex_search/broad_search.py:114
↓ 14 callersMethodmerged
(self, *collections: DocumentCollection)
libs/dcollection/ai2i/dcollection/interface/collection.py:283
↓ 12 callersMethodconfig
( self, reader: ConfigValuePlaceholder[A], /, default: B | ValueNotSet = Value
libs/di/ai2i/di/interface/gateway.py:66
↓ 12 callersFunctioncustom_gather
( pt1: _Promise[T1], pt2: _Promise[T2], /, *, return_exceptions: Literal[False] = False, )
agents/mabool/api/mabool/utils/asyncio.py:16
↓ 12 callersMethodgpt4
(**params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:234
↓ 11 callersMethodacquire
(self, priority: int = DEFAULT_PRIORITY)
agents/mabool/api/mabool/services/prioritized_task.py:41
↓ 11 callersMethodcontra_map
(self, f: Callable[[NEW_IN], IN], /, input_type: None = None)
libs/chain/ai2i/chain/computation.py:111
↓ 11 callersFunctiondefault_retry_settings
(logger: logging.Logger | None = None)
libs/chain/ai2i/chain/endpoints.py:68
↓ 11 callersMethodexecute
(self, computation: ChainComputation[IN, OUT])
libs/chain/ai2i/chain/endpoints.py:301
↓ 11 callersMethodfilter
(self, doc: dc.Document)
agents/mabool/api/mabool/agents/metadata_only/ops.py:442
↓ 11 callersMethodmerge
( collections: Iterable[DocumentCollection], dcf: DocumentCollectionFactory = DI.requires(dc_d
agents/mabool/api/mabool/utils/dc.py:44
↓ 11 callersMethodrelease
(self)
agents/mabool/api/mabool/services/prioritized_task.py:60
↓ 11 callersFunctionstore_value
(prefix: str, value: MyValue = DI.requires(my_value))
libs/di/ai2i/di/tests/test_managed_env.py:53
↓ 10 callersFunctionapplication_config_ctx
(app_conf: AppConfig)
libs/config/ai2i/config/config.py:60
↓ 10 callersFunctionget_relevant_docs
(documents: DocumentCollection, threshold: int | None = None)
agents/mabool/api/mabool/agents/common/relevance_judgement_utils.py:55
↓ 10 callersMethodregister
( self, model_name: str, delay_seconds: float = 0.0, return_value: str | None
libs/chain/ai2i/chain/tests/test_endpoints.py:110
↓ 10 callersFunctiontask
(order: List[int], index: int)
agents/mabool/api/mabool/services/tests/test_prioritized_task.py:20
↓ 9 callersMethod__init__
(self, msg: str)
libs/di/ai2i/di/interface/errors.py:27
↓ 9 callersFunctioncreate_module
(name: str, *, extends: Sequence[Module] = ())
libs/di/ai2i/di/factory/modules.py:7
↓ 9 callersFunctiondefine_llm_endpoint
( *, default_model: LLMModel[Any] | None = None, default_timeout: MaboolTimeout | None = None,
libs/chain/ai2i/chain/endpoints.py:330
↓ 9 callersMethodfrom_s2_search
( query: str, limit: int, search_iteration: int = 1, time_range: ExtractedYear
agents/mabool/api/mabool/utils/dc.py:77
↓ 9 callersMethodis_loaded
(self, e: str)
libs/dcollection/ai2i/dcollection/interface/document.py:179
↓ 9 callersMethodsorted
(self, sort_definitions: Sequence[DocumentCollectionSortDef])
libs/dcollection/ai2i/dcollection/interface/collection.py:306
↓ 8 callersMethod_norm
(c: specs.Composite[T])
agents/mabool/api/mabool/agents/metadata_only/plan.py:418
↓ 8 callersMethodfrom_docs
( documents: Sequence[Document], computed_fields: dict[DocumentFieldName, Any] | None = None,
agents/mabool/api/mabool/utils/dc.py:28
↓ 8 callersMethodis_loaded
(self, field_name: DocumentFieldName)
libs/dcollection/ai2i/dcollection/document.py:140
↓ 8 callersFunctionresolve_paper
(paper_spec: specs.PaperSpec, prev: Optional[ops.DocOp] = None)
agents/mabool/api/mabool/agents/metadata_only/plan.py:525
↓ 8 callersMethodtake
(self, n: int)
libs/dcollection/ai2i/dcollection/interface/collection.py:292
↓ 7 callersMethodask
(self, question: InquiryQuestion)
agents/mabool/api/mabool/infra/operatives/operatives.py:205
↓ 7 callersFunctionbatch_process
( items: Sequence[A], batch_size: int, process_func: Callable[[Sequence[A]], Awaitable[Sequence[B]
libs/common/ai2i/common/utils/batch.py:76
↓ 7 callersMethodclose_scope
(self)
libs/di/ai2i/di/interface/scopes.py:51
↓ 7 callersMethodcreate_task
(self, coro: Coroutine[Any, Any, A])
libs/di/ai2i/di/interface/tasks.py:8
↓ 7 callersFunctiondependency_topological_sort
Topological sort of dependency definitions based on their defined dependencies This will dictate the initialization order of the definitions
libs/di/ai2i/di/utils.py:8
↓ 7 callersMethoddimap
(self, f: Callable[[NEW_IN], IN], g: Callable[[OUT], NEW_OUT])
libs/chain/ai2i/chain/computation.py:98
↓ 7 callersMethodfrom_docs
Create a document collection from a list of documents.
libs/dcollection/ai2i/dcollection/factory.py:82
↓ 7 callersFunctionget_fields_of_study_filter_from_domains
(domains: DomainsIdentified)
agents/mabool/api/mabool/agents/common/domain_utils.py:30
↓ 7 callersMethodopen_scope
(self, values: CachedValues | None = None)
libs/di/ai2i/di/interface/scopes.py:48
↓ 6 callersMethod_get_default_llm_params
()
libs/chain/ai2i/chain/models.py:190
↓ 6 callersFunctioncamel_case
(s: str)
libs/config/ai2i/config/gen_config_types.py:28
↓ 6 callersFunctionextract_specifications
( user_input: str, )
agents/mabool/api/mabool/agents/query_analyzer/query_analyzer.py:251
↓ 6 callersFunctionfunc
( value: str = DI.requires(my_value), )
libs/di/ai2i/di/tests/test_managed_env.py:396
↓ 6 callersMethodis_non_trivial_metadata_only
Check if the specification is non-trivial but metadata-only.
agents/mabool/api/mabool/data_model/specifications.py:197
↓ 6 callersMethodmerge
(self, collections: Iterable[DocumentCollection])
libs/dcollection/ai2i/dcollection/factory.py:108
↓ 6 callersMethodnon_empty
(self)
libs/dcollection/ai2i/dcollection/interface/document.py:23
↓ 6 callersMethodpipe_to
(self, p: ChainComputation[OUT, NEW_OUT])
libs/chain/ai2i/chain/computation.py:156
↓ 5 callersMethod__aexit__
( self, exc_type: Optional[type[BaseException]], exc_val: Optional[BaseException], exc_tb: Optional[An
agents/mabool/api/mabool/services/prioritized_task.py:74
↓ 5 callersFunction_create_doc_loading_error
( doc_id: CorpusId, reason: str, original_exception: Exception | None = None )
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:386
↓ 5 callersMethodainvoke
(self, input: IN, config: RunnableConfig | None = None, **kwargs: Any)
libs/chain/ai2i/chain/retry.py:61
↓ 5 callersMethodclose_round_scope
(self)
libs/di/ai2i/di/interface/app_context.py:50
↓ 5 callersMethodcreate_factory
(self)
libs/chain/ai2i/chain/tests/test_endpoints.py:128
↓ 5 callersFunctioncustom_gather
( pt1: _Promise[T1], pt2: _Promise[T2], /, *, return_exceptions: Literal[False] = False,
libs/common/ai2i/common/utils/asyncio.py:12
↓ 5 callersFunctiondummy_task
()
agents/mabool/api/mabool/services/tests/test_prioritized_task.py:38
↓ 5 callersMethodfields
(self)
libs/dcollection/ai2i/dcollection/document.py:129
↓ 5 callersMethodfrom_function
(f: Callable[P2, B2])
libs/config/ai2i/config/common.py:29
↓ 5 callersMethodfrom_ids
( corpus_ids: list[CorpusId], dcf: DocumentCollectionFactory = DI.requires(dc_deps.round_doc_collectio
agents/mabool/api/mabool/utils/dc.py:21
↓ 5 callersFunctionget_paper_finder_output_from_docs
( docs: DocumentCollection, aggregated_response_text: str, user_query: str, analyzed_input: An
agents/mabool/api/mabool/agents/paper_finder/paper_finder_agent.py:507
↓ 5 callersMethodgpt41mini
(**params: Unpack[LLMModelParams])
libs/chain/ai2i/chain/models.py:214
↓ 5 callersMethodis_empty
Check if the object is empty (all fields are None or empty).
agents/mabool/api/mabool/data_model/specifications.py:49
↓ 5 callersFunctionload_conf
(conf_dir: Path)
libs/config/ai2i/config/loading.py:65
↓ 5 callersMethodpassthrough_input
(self)
libs/chain/ai2i/chain/computation.py:125
↓ 5 callersFunctionrelevance_judgement_field
( relevance_criteria: RelevanceCriteria, field_name: str = "relevance_judgement" )
agents/mabool/api/mabool/agents/common/computed_fields/relevance.py:74
↓ 5 callersMethodresolve
(self, definition: DependencyDefinition[A])
libs/di/ai2i/di/scopes.py:354
↓ 5 callersFunctions2_paper_to_document
( corpus_id: str, paper: Paper | None = None, origin_query: OriginQuery | None = None, context
libs/dcollection/ai2i/dcollection/loaders/s2_rest.py:218
↓ 5 callersFunctionsplit_snippets_to_sentence_snippets
(snippet: Snippet)
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:224
↓ 5 callersFunctionsplit_text_to_sentence_offsets
(snippet: Snippet)
agents/mabool/api/mabool/agents/snowball/snippet_snowball.py:151
↓ 4 callersMethod__aenter__
(self)
agents/mabool/api/mabool/services/prioritized_task.py:70
↓ 4 callersFunction_accumulate_snippet_matches
Reusable helper that finds snippet parts in a given text and accumulates them. The `factory` callback should produce a Snippet or CitationCon
agents/mabool/api/mabool/agents/common/computed_fields/relevant_snippets.py:86
↓ 4 callersMethodclose_scope
(self)
libs/di/ai2i/di/scopes.py:110
↓ 4 callersMethodcontext
(self)
libs/dcollection/ai2i/dcollection/factory.py:75
↓ 4 callersFunctiondefine_parser_with_llm
( parser: PydanticOutputParser[LLM_STRUCTURED_OUTPUT], )
libs/chain/ai2i/chain/builders.py:295
↓ 4 callersMethodempty
()
libs/config/ai2i/config/config_models.py:39
next →1–100 of 1,349, ranked by callers