Files
qdrant-client/tools/async_client_generator/base_generator.py
Andrey Vasnetsov 0f32e578ec update interface and version for fastembed (#340)
* update interface and version for fastembed

* fix types

* fix types

* regen async

* use python 3.11 to check compatibility

* fix docstring

* regen async

* propagate batch size
2023-10-16 18:13:52 +02:00

18 lines
449 B
Python

import ast
from typing import List
from tools.async_client_generator.config import AUTOGEN_WARNING_MESSAGE
class BaseGenerator:
def __init__(self) -> None:
self.transformers: List[ast.NodeTransformer] = []
def generate(self, code: str) -> str:
nodes = ast.parse(code)
for transformer in self.transformers:
nodes = transformer.visit(nodes)
return AUTOGEN_WARNING_MESSAGE + ast.unparse(nodes)