From 619b855d5fb203b31b9d4c731b988169f7aef256 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Fri, 25 Dec 2020 13:11:01 +0100
Subject: [PATCH] Add version endpoint

---
 freqtrade/rpc/api_server2/api_v1.py | 17 ++++++++++++-----
 freqtrade/rpc/api_server2/models.py |  7 +++----
 2 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/freqtrade/rpc/api_server2/api_v1.py b/freqtrade/rpc/api_server2/api_v1.py
index 2d1491b06..767d72023 100644
--- a/freqtrade/rpc/api_server2/api_v1.py
+++ b/freqtrade/rpc/api_server2/api_v1.py
@@ -1,12 +1,14 @@
-from typing import Dict
-
 from fastapi import APIRouter, Depends
 
-from .deps import get_rpc, get_config
-from .models import Balances, Ping
+from freqtrade import __version__
+
+from .deps import get_config, get_rpc
+from .models import Balances, Ping, Version
+
 
 # Public API, requires no auth.
 router_public = APIRouter()
+# Private API, protected by authentication
 router = APIRouter()
 
 
@@ -17,5 +19,10 @@ def ping():
 
 
 @router.get('/balance', response_model=Balances)
-def balance(rpc=Depends(get_rpc), config=Depends(get_config)) -> Dict[str, str]:
+def balance(rpc=Depends(get_rpc), config=Depends(get_config)):
     return rpc._rpc_balance(config['stake_currency'], config.get('fiat_display_currency', ''),)
+
+
+@router.get('/version', response_model=Version)
+def version():
+    return {"version": __version__}
diff --git a/freqtrade/rpc/api_server2/models.py b/freqtrade/rpc/api_server2/models.py
index 3d1fbf969..7cd628da0 100644
--- a/freqtrade/rpc/api_server2/models.py
+++ b/freqtrade/rpc/api_server2/models.py
@@ -5,10 +5,9 @@ from pydantic import BaseModel
 class Ping(BaseModel):
     status: str
 
-    class Config:
-        schema_extra = {
-            "example": {"status", "pong"}
-        }
+
+class Version(BaseModel):
+    version: str
 
 
 class Balance(BaseModel):