mirror of
https://github.com/qdrant/qdrant-client.git
synced 2026-08-03 16:40:59 -05:00
278a68d297
* new: manually implemented async qdrant client * fix: remove await before sync call * fix: make upload collection, records and migrate synchronous * fix: add init method to async client base * refactoring: remove redundant import * new: add super().__init__ in qdrant remote, update import in http * new: mvp async qdrant client generator * new: fix mypy, update generator script, refactoring * fix: fix test script * new: update generator launch script, update async files * new: refactor async client generator * refactoring: remove redundant operations, add comments, refactor * new: add isort, black and autoflake to dev dependencies * fix: add more checks, fix type hints * fix: do not check types in async client generator for python3.8 * new: do not type check async_qdrant_fastembed * fix: fix pyright run * new: update async client tests * fix: update versions in CI * new: update pre-commit python version, update autogenerated files * new: update generated files, add tests for async generator, update generator script * fix: exclude generator test from 3.8 * fix: fix python version condition * fix: add python target version for black * fix: generate async client only on python 3.9
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from typing import Dict, List, Optional
|
|
|
|
from tools.async_client_generator.base_generator import BaseGenerator
|
|
from tools.async_client_generator.transformers import (
|
|
ClassDefTransformer,
|
|
ConstantTransformer,
|
|
FunctionDefTransformer,
|
|
)
|
|
|
|
|
|
class BaseClientGenerator(BaseGenerator):
|
|
def __init__(
|
|
self,
|
|
keep_sync: Optional[List[str]] = None,
|
|
class_replace_map: Optional[Dict[str, str]] = None,
|
|
constant_replace_map: Optional[Dict[str, str]] = None,
|
|
):
|
|
super().__init__()
|
|
|
|
self.transformers.append(FunctionDefTransformer(keep_sync=keep_sync))
|
|
self.transformers.append(ClassDefTransformer(class_replace_map=class_replace_map))
|
|
self.transformers.append(ConstantTransformer(constant_replace_map=constant_replace_map))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from tools.async_client_generator.config import CLIENT_DIR, CODE_DIR
|
|
|
|
with open(CLIENT_DIR / "client_base.py", "r") as source_file:
|
|
code = source_file.read()
|
|
|
|
# Parse the code into an AST
|
|
base_client_generator = BaseClientGenerator(
|
|
keep_sync=["__init__", "upload_records", "upload_collection", "migrate"],
|
|
class_replace_map={"QdrantBase": "AsyncQdrantBase"},
|
|
constant_replace_map={"QdrantBase": "AsyncQdrantBase"},
|
|
)
|
|
modified_code = base_client_generator.generate(code)
|
|
|
|
with open(CODE_DIR / "async_client_base.py", "w") as target_file:
|
|
target_file.write(modified_code)
|