main.py
5.2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
#!/usr/bin/env python3
"""
Main entry point for SearchEngine operations.
Provides a unified CLI for common operations:
- serve: Start API service (search + admin routes)
- serve-indexer: Start dedicated Indexer API service
- search: Test search from command line
"""
import sys
import os
import argparse
import json
import uvicorn
# Add parent directory to path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from config import ConfigLoader
from utils import ESClient
from search import Searcher
def cmd_serve(args):
"""Start API service."""
os.environ['ES_HOST'] = args.es_host
print("Starting API service (multi-tenant)...")
print(f" Host: {args.host}:{args.port} (search + indexer routes)")
print(f" Elasticsearch: {args.es_host}")
uvicorn.run(
"api.app:app",
host=args.host,
port=args.port,
reload=args.reload
)
def cmd_serve_indexer(args):
"""Start dedicated Indexer API service (no search endpoints)."""
os.environ['ES_HOST'] = args.es_host
print("Starting Indexer API service...")
print(f" Host: {args.host}:{args.port} (indexer only)")
print(f" Elasticsearch: {args.es_host}")
uvicorn.run(
"api.indexer_app:app",
host=args.host,
port=args.port,
reload=args.reload
)
def cmd_search(args):
"""Test search from command line."""
# Load config
config_loader = ConfigLoader("config/config.yaml")
config = config_loader.load_config()
# Initialize ES and searcher
es_client = ESClient(hosts=[args.es_host])
if not es_client.ping():
print(f"ERROR: Cannot connect to Elasticsearch at {args.es_host}")
return 1
from query import QueryParser
query_parser = QueryParser(config)
searcher = Searcher(es_client, config, query_parser)
# Execute search
print(f"Searching for: '{args.query}' (tenant: {args.tenant_id})")
result = searcher.search(
query=args.query,
tenant_id=args.tenant_id,
size=args.size
)
# Display results
print(f"\nFound {result.total} results in {result.took_ms}ms")
print(f"Max score: {result.max_score}")
if args.json:
print(json.dumps(result.to_dict(), indent=2, ensure_ascii=False))
else:
print(f"\nTop {len(result.hits)} results:")
for i, hit in enumerate(result.hits, 1):
source = hit['_source']
score = hit['_score']
print(f"\n{i}. Score: {score:.4f}")
print(f" ID: {source.get('skuId', 'N/A')}")
print(f" Name: {source.get('name', 'N/A')}")
print(f" Category: {source.get('categoryName', 'N/A')}")
print(f" Brand: {source.get('brandName', 'N/A')}")
return 0
def main():
"""Main CLI entry point."""
parser = argparse.ArgumentParser(
description='SearchEngine - E-Commerce Search SaaS',
formatter_class=argparse.RawDescriptionHelpFormatter
)
subparsers = parser.add_subparsers(dest='command', help='Command to execute')
# Serve command
serve_parser = subparsers.add_parser('serve', help='Start API service (multi-tenant)')
serve_parser.add_argument('--host', default='0.0.0.0', help='Host to bind to')
serve_parser.add_argument('--port', type=int, default=6002, help='Port to bind to')
serve_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
serve_parser.add_argument('--reload', action='store_true', help='Enable auto-reload')
# Serve-indexer command
serve_indexer_parser = subparsers.add_parser(
'serve-indexer',
help='Start dedicated Indexer API service (indexer routes only)'
)
serve_indexer_parser.add_argument('--host', default='0.0.0.0', help='Host to bind to')
serve_indexer_parser.add_argument('--port', type=int, default=6004, help='Port to bind to')
serve_indexer_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
serve_indexer_parser.add_argument('--reload', action='store_true', help='Enable auto-reload')
# Search command
search_parser = subparsers.add_parser('search', help='Test search from command line')
search_parser.add_argument('query', help='Search query')
search_parser.add_argument('--tenant-id', required=True, help='Tenant ID (required)')
search_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
search_parser.add_argument('--size', type=int, default=10, help='Number of results')
search_parser.add_argument('--no-translation', action='store_true', help='Disable translation')
search_parser.add_argument('--no-embedding', action='store_true', help='Disable embeddings')
search_parser.add_argument('--json', action='store_true', help='Output JSON')
args = parser.parse_args()
if not args.command:
parser.print_help()
return 1
# Execute command
if args.command == 'serve':
return cmd_serve(args)
elif args.command == 'serve-indexer':
return cmd_serve_indexer(args)
elif args.command == 'search':
return cmd_search(args)
else:
print(f"Unknown command: {args.command}")
return 1
if __name__ == "__main__":
sys.exit(main())