refactor: stock_evaluate 요청 및 응답 형식 변경

aiServer/stock/ai_model.py

@@ -1,26 +1,26 @@
-from array import array
-from typing import List, Dict, Any
+from typing import Any
 
 
 class Output:
-    def __init__(self, product_number, name, percentage):
+    def __init__(self, product_number, name, quantity):
         self.product_number = product_number
         self.name = name
-        self.percentage = percentage
+        self.quantity = quantity
 
     def to_dict(self):
         return {
             'productNumber': self.product_number,
             'name': self.name,
-            "percentage": self.percentage,
+            "quantity": self.quantity,
         }
 
 
 # 요청된 주식 목록에 따라 각 주식의 구매 비율을 리턴하는 메서드
 def get_stock_order_ratio(stocks) -> list[dict[str, Any]]:
+    print(stocks)
 
     outputs = []
-    output = Output("삼성전자", "000000", 0.1)
+    output = Output("005930", "삼성전자", 1)
     outputs.append(output.to_dict())
 
     return outputs

aiServer/stock/views.py

@@ -12,7 +12,8 @@ def recommended_stock_list(request):
 @api_view(['POST'])
 def stock_evaluate(request) -> JsonResponse:
     stocks = request.data['stocks']
+    amount = request.data['amount']  # 사용 가능 금액
 
     response = get_stock_order_ratio(stocks)
 
-    return JsonResponse({"outputs": response}, safe=True)
+    return JsonResponse({"stocks": response}, safe=True)