refactor: 重构 algorithm 为 ai-core 代码解析服务
- 新增 ai-core 目录,包含代码解析核心服务 - 添加 proto 定义、parser、service 模块 - 添加启动脚本和依赖配置 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
46
ai-core/generate_grpc.py
Normal file
46
ai-core/generate_grpc.py
Normal file
@@ -0,0 +1,46 @@
|
||||
import subprocess
|
||||
import sys
|
||||
import os
|
||||
|
||||
proto_file = "proto/document_parser.proto"
|
||||
proto_path = "proto"
|
||||
python_out = "proto"
|
||||
grpc_python_out = "proto"
|
||||
|
||||
def generate_grpc():
|
||||
"""Generate gRPC Python code from proto file"""
|
||||
print(f"Generating gRPC code from {proto_file}...")
|
||||
|
||||
cmd = [
|
||||
sys.executable,
|
||||
"-m",
|
||||
"grpc_tools.protoc",
|
||||
f"--proto_path={proto_path}",
|
||||
f"--python_out={python_out}",
|
||||
f"--grpc_python_out={grpc_python_out}",
|
||||
proto_file,
|
||||
]
|
||||
|
||||
try:
|
||||
subprocess.run(cmd, check=True)
|
||||
print("gRPC code generated successfully!")
|
||||
|
||||
pb2_file = os.path.join(python_out, "document_parser_pb2.py")
|
||||
pb2_grpc_file = os.path.join(python_out, "document_parser_pb2_grpc.py")
|
||||
|
||||
if os.path.exists(pb2_file) and os.path.exists(pb2_grpc_file):
|
||||
print(f"Generated files:")
|
||||
print(f" - {pb2_file}")
|
||||
print(f" - {pb2_grpc_file}")
|
||||
else:
|
||||
print("Warning: Expected files not found")
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"Error generating gRPC code: {e}")
|
||||
sys.exit(1)
|
||||
except Exception as e:
|
||||
print(f"Unexpected error: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
generate_grpc()
|
||||
Reference in New Issue
Block a user