Skip to content

V2 Arrow Clients #918

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 6 commits into from
Jul 23, 2025
Merged

V2 Arrow Clients #918

merged 6 commits into from
Jul 23, 2025

Conversation

DarthMax
Copy link
Contributor

@DarthMax DarthMax commented Jul 17, 2025

Part of GDSA-109

Copy link

netlify bot commented Jul 17, 2025

Deploy Preview for neo4j-graph-data-science-client canceled.

Name Link
🔨 Latest commit 5b7e29f
🔍 Latest deploy log https://app.netlify.com/projects/neo4j-graph-data-science-client/deploys/68809b73b53235000864415d

@DarthMax DarthMax requested a review from Copilot July 17, 2025 18:11
Copy link

@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR introduces V2 Arrow Clients for the GraphDataScience library, implementing a new Arrow-based communication layer with authentication, middleware, and client functionality for job management, mutations, and write-back operations.

  • Adds authentication middleware with token management and basic auth support
  • Implements V2 client classes for job management, mutations, and write-back operations
  • Introduces data models and utilities for Arrow Flight communication

Reviewed Changes

Copilot reviewed 19 out of 24 changed files in this pull request and generated 4 comments.

Show a summary per file
File Description
mypy.ini Adds pydantic.mypy plugin for type checking
graphdatascience/arrow_client/arrow_authentication.py Defines authentication interfaces and implementations
graphdatascience/arrow_client/arrow_base_model.py Base model class with camelCase serialization
graphdatascience/arrow_client/arrow_info.py Arrow server information data class
graphdatascience/arrow_client/authenticated_flight_client.py Main authenticated Arrow Flight client implementation
graphdatascience/arrow_client/middleware/auth_middleware.py Authentication middleware for handling tokens and basic auth
graphdatascience/arrow_client/middleware/user_agent_middleware.py User agent middleware for request headers
graphdatascience/arrow_client/v2/api_types.py API type definitions for V2 clients
graphdatascience/arrow_client/v2/data_mapper_utils.py Utilities for deserializing Arrow results
graphdatascience/arrow_client/v2/job_client.py Client for job management operations
graphdatascience/arrow_client/v2/mutation_client.py Client for mutation operations
graphdatascience/arrow_client/v2/write_back_client.py Client for write-back operations
Test files Unit tests for all new client functionality

@DarthMax DarthMax merged commit 713cc5c into neo4j:main Jul 23, 2025
8 checks passed
@DarthMax DarthMax deleted the v2_arrow_clients branch July 23, 2025 12:16
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.

1 participant