Skip to content

feat: galileo adk to natively support custom retriever tools#482

Merged
savula15 merged 1 commit intomainfrom
adk-retriever-support
Feb 22, 2026
Merged

feat: galileo adk to natively support custom retriever tools#482
savula15 merged 1 commit intomainfrom
adk-retriever-support

Conversation

@savula15
Copy link
Contributor

@savula15 savula15 commented Feb 19, 2026

User description

Shortcut:
https://app.shortcut.com/galileo/story/55769/galileo-adk-to-support-retriever-tools

Tests:

  • Unit Tests Added
  • E2E Test Added (if it's a user-facing feature, or fixing a bug)

Generated description

Below is a concise technical summary of the changes proposed in this PR:
Introduces native support for custom retriever tools, enabling RAG quality metrics in Galileo by providing a @galileo_retriever decorator and enhancing the GalileoObserver to detect and log retriever-specific spans. These changes ensure that retrieval operations are correctly identified and their outputs are transformed into a standardized document format for comprehensive observability.

TopicDetails
Retriever Span Processing Enhances the SpanManager to create distinct 'retriever' node types with specific tags and modifies the TraceBuilder to convert various retriever output formats into a standardized list of documents, crucial for RAG quality metric analysis. This ensures accurate logging and data processing for retriever spans.
Modified files (4)
  • galileo-adk/src/galileo_adk/span_manager.py
  • galileo-adk/src/galileo_adk/trace_builder.py
  • galileo-adk/tests/test_span_manager.py
  • galileo-adk/tests/test_trace_builder.py
Latest Contributors(1)
UserCommitDate
sankar@galileo.aifeat-add-galileo-adk-i...February 10, 2026
Retriever Detection API Implements the @galileo_retriever decorator to mark functions as retrievers and updates the GalileoObserver to detect these tools, either through the decorator's attribute or by checking for BaseRetrievalTool inheritance. This includes exposing the decorator in the public API and providing user documentation.
Modified files (5)
  • galileo-adk/README.md
  • galileo-adk/src/galileo_adk/__init__.py
  • galileo-adk/src/galileo_adk/decorator.py
  • galileo-adk/src/galileo_adk/observer.py
  • galileo-adk/tests/test_retriever.py
Latest Contributors(2)
UserCommitDate
ci@rungalileo.iochore-release-galileo-...February 11, 2026
sankar@galileo.aifeat-add-galileo-adk-i...February 10, 2026
Project Configuration Updates project metadata, Python version classifiers, and the galileo dependency to version 1.47.0 in pyproject.toml, along with adding a repository URL. These are foundational changes for project compatibility and discoverability.
Modified files (1)
  • galileo-adk/pyproject.toml
Latest Contributors(2)
UserCommitDate
ci@rungalileo.iochore-release-galileo-...February 11, 2026
sankar@galileo.aifeat-add-galileo-adk-i...February 10, 2026
This pull request is reviewed by Baz. Review like a pro on (Baz).

@savula15 savula15 requested a review from a team as a code owner February 19, 2026 00:05
@codecov
Copy link

codecov bot commented Feb 19, 2026

Codecov Report

❌ Patch coverage is 93.33333% with 2 lines in your changes missing coverage. Please review.
✅ Project coverage is 82.57%. Comparing base (a3106fd) to head (3de8546).
⚠️ Report is 1 commits behind head on main.

Files with missing lines Patch % Lines
galileo-adk/src/galileo_adk/observer.py 84.61% 2 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main     #482      +/-   ##
==========================================
+ Coverage   81.85%   82.57%   +0.72%     
==========================================
  Files          96      106      +10     
  Lines        9164    10269    +1105     
==========================================
+ Hits         7501     8480     +979     
- Misses       1663     1789     +126     
Flag Coverage Δ
galileo-adk 88.59% <93.33%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@savula15 savula15 enabled auto-merge (squash) February 19, 2026 00:14
@savula15 savula15 force-pushed the adk-retriever-support branch from 8ec9edd to 50a5891 Compare February 19, 2026 00:34
@savula15 savula15 force-pushed the adk-retriever-support branch from 3937b2b to 1693801 Compare February 21, 2026 03:20
@savula15 savula15 force-pushed the adk-retriever-support branch from 1693801 to 52d274a Compare February 21, 2026 03:30
@savula15 savula15 force-pushed the adk-retriever-support branch from 52d274a to 3de8546 Compare February 21, 2026 03:31
@savula15 savula15 merged commit f11138a into main Feb 22, 2026
54 of 56 checks passed
@savula15 savula15 deleted the adk-retriever-support branch February 22, 2026 00:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants