matsuap commited on
Commit
526ac59
·
1 Parent(s): 1a2f833

住所比較および正規化機能のエンドポイントを更新し、リクエストからJSONデータを取得するように修正。これにより、住所1、住所2、およびクエリ住所の処理が改善されました。

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -10,7 +10,7 @@ import time
10
  import requests
11
  import pandas as pd
12
  import os
13
- from fastapi import FastAPI
14
  from pymilvus import MilvusClient
15
  from dotenv import load_dotenv
16
  import time
@@ -373,7 +373,11 @@ def get_addresses_with_rsdtdsp(pref, county, city, ward, oaza_cho, chome, koaza)
373
  app = FastAPI()
374
 
375
  @app.post("/compare-two-addresses")
376
- def compare_two_addresses(address1, address2):
 
 
 
 
377
  preprocessed1 = preprocess(address1)
378
  preprocessed2 = preprocess(address2)
379
  hits1 = vector_search(preprocessed1, top_k=1)
@@ -384,7 +388,10 @@ def compare_two_addresses(address1, address2):
384
  return result
385
 
386
  @app.post("/normalize-address")
387
- def normalize_address(query_address):
 
 
 
388
  with measure('preprocess'):
389
  preprocessed = preprocess(query_address)
390
  with measure('vector_search'):
 
10
  import requests
11
  import pandas as pd
12
  import os
13
+ from fastapi import FastAPI, Request
14
  from pymilvus import MilvusClient
15
  from dotenv import load_dotenv
16
  import time
 
373
  app = FastAPI()
374
 
375
  @app.post("/compare-two-addresses")
376
+ def compare_two_addresses(request: Request):
377
+ data = request.json()
378
+ address1 = data.get('address1')
379
+ address2 = data.get('address2')
380
+
381
  preprocessed1 = preprocess(address1)
382
  preprocessed2 = preprocess(address2)
383
  hits1 = vector_search(preprocessed1, top_k=1)
 
388
  return result
389
 
390
  @app.post("/normalize-address")
391
+ def normalize_address(request: Request):
392
+ data = request.json()
393
+ query_address = data.get('query_address')
394
+
395
  with measure('preprocess'):
396
  preprocessed = preprocess(query_address)
397
  with measure('vector_search'):