-
- Downloads
Add `span_id` attribute to Events (instrumentation) (#12417)
* add span_id to Event * remove raise err in NullHandler * wip * modify root dispatcher event enclosing span * remove *args as we have bound_args now * add LLMChatInProgressEvent * add LLMStructuredPredict Eventst * store span_id before await executions * add SpanDropEvent with err_str payload * add event to _achat; flush current_span_id when open_spans is empty * llm callbacks use root span_id * add unit tests * remove print statements * provide context manager returning a distpatch event partial with correct span id * move to context manager usage * fix invocation of cm * define and use get_dispatch_event method * remove aim tests
Showing
- llama-index-core/llama_index/core/agent/runner/base.py 18 additions, 5 deletionsllama-index-core/llama_index/core/agent/runner/base.py
- llama-index-core/llama_index/core/base/base_query_engine.py 8 additions, 4 deletionsllama-index-core/llama_index/core/base/base_query_engine.py
- llama-index-core/llama_index/core/base/base_retriever.py 24 additions, 6 deletionsllama-index-core/llama_index/core/base/base_retriever.py
- llama-index-core/llama_index/core/base/embeddings/base.py 74 additions, 16 deletionsllama-index-core/llama_index/core/base/embeddings/base.py
- llama-index-core/llama_index/core/chat_engine/types.py 18 additions, 8 deletionsllama-index-core/llama_index/core/chat_engine/types.py
- llama-index-core/llama_index/core/instrumentation/__init__.py 4 additions, 1 deletion...a-index-core/llama_index/core/instrumentation/__init__.py
- llama-index-core/llama_index/core/instrumentation/dispatcher.py 83 additions, 9 deletions...index-core/llama_index/core/instrumentation/dispatcher.py
- llama-index-core/llama_index/core/instrumentation/events/base.py 1 addition, 0 deletions...ndex-core/llama_index/core/instrumentation/events/base.py
- llama-index-core/llama_index/core/instrumentation/events/llm.py 24 additions, 0 deletions...index-core/llama_index/core/instrumentation/events/llm.py
- llama-index-core/llama_index/core/instrumentation/events/span.py 10 additions, 0 deletions...ndex-core/llama_index/core/instrumentation/events/span.py
- llama-index-core/llama_index/core/instrumentation/span/simple.py 2 additions, 1 deletion...ndex-core/llama_index/core/instrumentation/span/simple.py
- llama-index-core/llama_index/core/instrumentation/span_handlers/base.py 9 additions, 7 deletions...re/llama_index/core/instrumentation/span_handlers/base.py
- llama-index-core/llama_index/core/instrumentation/span_handlers/null.py 1 addition, 7 deletions...re/llama_index/core/instrumentation/span_handlers/null.py
- llama-index-core/llama_index/core/instrumentation/span_handlers/simple.py 8 additions, 10 deletions.../llama_index/core/instrumentation/span_handlers/simple.py
- llama-index-core/llama_index/core/llms/callbacks.py 33 additions, 4 deletionsllama-index-core/llama_index/core/llms/callbacks.py
- llama-index-core/llama_index/core/llms/llm.py 24 additions, 5 deletionsllama-index-core/llama_index/core/llms/llm.py
- llama-index-core/llama_index/core/response_synthesizers/base.py 50 additions, 14 deletions...index-core/llama_index/core/response_synthesizers/base.py
- llama-index-core/llama_index/core/response_synthesizers/refine.py 8 additions, 3 deletions...dex-core/llama_index/core/response_synthesizers/refine.py
- llama-index-core/tests/instrumentation/test_dispatcher.py 218 additions, 24 deletionsllama-index-core/tests/instrumentation/test_dispatcher.py
- llama-index-integrations/callbacks/llama-index-callbacks-aim/tests/BUILD 0 additions, 1 deletion...egrations/callbacks/llama-index-callbacks-aim/tests/BUILD
Loading
Please register or sign in to comment