cnhkmcp 2.0.2__tar.gz → 2.0.3__tar.gz

{cnhkmcp-2.0.2/cnhkmcp.egg-info → cnhkmcp-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 2.0.2
+Version: 2.0.3
 Summary: A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration
 Home-page: https://github.com/cnhk/cnhkmcp
 Author: CNHK

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3}/cnhkmcp/untracked/APP/Tranformer/ace_lib.py

@@ -1134,7 +1134,7 @@ def get_datasets(
     region: str = "USA",
     delay: int = 1,
     universe: str = "TOP3000",
-    theme: str = "false",
+    theme: str = "ALL",
 ) -> pd.DataFrame:
     """
     Retrieve available datasets based on specified parameters.
@@ -1145,18 +1145,39 @@ def get_datasets(
         region (str, optional): The region. Defaults to "USA".
         delay (int, optional): The delay. Defaults to 1.
         universe (str, optional): The universe. Defaults to "TOP3000".
-        theme (str, optional): The theme. Defaults to "false".
+        theme (str, optional): The theme. Defaults to "ALL".
 
     Returns:
         pandas.DataFrame: A DataFrame containing information about available datasets.
     """
-    url = (
-        brain_api_url
-        + "/data-sets?"
-        + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
-    )
-    result = s.get(url)
-    datasets_df = pd.DataFrame(result.json()["results"])
+    if theme == "ALL":
+        # Fetch both theme=false and theme=true
+        url_false = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=false"
+        )
+        result_false = s.get(url_false)
+        df_false = pd.DataFrame(result_false.json()["results"])
+        
+        url_true = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=true"
+        )
+        result_true = s.get(url_true)
+        df_true = pd.DataFrame(result_true.json()["results"])
+        
+        datasets_df = pd.concat([df_false, df_true], ignore_index=True)
+    else:
+        url = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
+        )
+        result = s.get(url)
+        datasets_df = pd.DataFrame(result.json()["results"])
+        
     datasets_df = expand_dict_columns(datasets_df)
     return datasets_df
 

{cnhkmcp-2.0.2/cnhkmcp/untracked/APP/give_me_idea → cnhkmcp-2.0.3/cnhkmcp/untracked/APP}/ace_lib.py

@@ -1134,7 +1134,7 @@ def get_datasets(
     region: str = "USA",
     delay: int = 1,
     universe: str = "TOP3000",
-    theme: str = "false",
+    theme: str = "ALL",
 ) -> pd.DataFrame:
     """
     Retrieve available datasets based on specified parameters.
@@ -1145,18 +1145,39 @@ def get_datasets(
         region (str, optional): The region. Defaults to "USA".
         delay (int, optional): The delay. Defaults to 1.
         universe (str, optional): The universe. Defaults to "TOP3000".
-        theme (str, optional): The theme. Defaults to "false".
+        theme (str, optional): The theme. Defaults to "ALL".
 
     Returns:
         pandas.DataFrame: A DataFrame containing information about available datasets.
     """
-    url = (
-        brain_api_url
-        + "/data-sets?"
-        + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
-    )
-    result = s.get(url)
-    datasets_df = pd.DataFrame(result.json()["results"])
+    if theme == "ALL":
+        # Fetch both theme=false and theme=true
+        url_false = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=false"
+        )
+        result_false = s.get(url_false)
+        df_false = pd.DataFrame(result_false.json()["results"])
+        
+        url_true = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=true"
+        )
+        result_true = s.get(url_true)
+        df_true = pd.DataFrame(result_true.json()["results"])
+        
+        datasets_df = pd.concat([df_false, df_true], ignore_index=True)
+    else:
+        url = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
+        )
+        result = s.get(url)
+        datasets_df = pd.DataFrame(result.json()["results"])
+        
     datasets_df = expand_dict_columns(datasets_df)
     return datasets_df
 

{cnhkmcp-2.0.2/cnhkmcp/untracked/APP → cnhkmcp-2.0.3/cnhkmcp/untracked/APP/give_me_idea}/ace_lib.py

@@ -1134,7 +1134,7 @@ def get_datasets(
     region: str = "USA",
     delay: int = 1,
     universe: str = "TOP3000",
-    theme: str = "false",
+    theme: str = "ALL",
 ) -> pd.DataFrame:
     """
     Retrieve available datasets based on specified parameters.
@@ -1145,18 +1145,39 @@ def get_datasets(
         region (str, optional): The region. Defaults to "USA".
         delay (int, optional): The delay. Defaults to 1.
         universe (str, optional): The universe. Defaults to "TOP3000".
-        theme (str, optional): The theme. Defaults to "false".
+        theme (str, optional): The theme. Defaults to "ALL".
 
     Returns:
         pandas.DataFrame: A DataFrame containing information about available datasets.
     """
-    url = (
-        brain_api_url
-        + "/data-sets?"
-        + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
-    )
-    result = s.get(url)
-    datasets_df = pd.DataFrame(result.json()["results"])
+    if theme == "ALL":
+        # Fetch both theme=false and theme=true
+        url_false = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=false"
+        )
+        result_false = s.get(url_false)
+        df_false = pd.DataFrame(result_false.json()["results"])
+        
+        url_true = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=true"
+        )
+        result_true = s.get(url_true)
+        df_true = pd.DataFrame(result_true.json()["results"])
+        
+        datasets_df = pd.concat([df_false, df_true], ignore_index=True)
+    else:
+        url = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
+        )
+        result = s.get(url)
+        datasets_df = pd.DataFrame(result.json()["results"])
+        
     datasets_df = expand_dict_columns(datasets_df)
     return datasets_df
 

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3}/cnhkmcp/untracked/APP/give_me_idea/alpha_data_specific_template_master.py

@@ -189,7 +189,11 @@ def main():
             universe=universe
         )
         print(f"Retrieved {len(datasets_df)} datasets.")
-        print(datasets_df[['id', 'name', 'category', 'subcategory']].head(10))
+        # print(datasets_df[['id', 'name', 'category', 'subcategory']].head(10))
+        
+        # Print all datasets for user selection
+        pd.set_option('display.max_rows', None)
+        print(datasets_df[['id', 'name', 'category', 'subcategory']])
     except Exception as e:
         print(f"Failed to get datasets: {e}")
         return

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3}/cnhkmcp/untracked/APP/hkSimulator/ace_lib.py

@@ -1143,7 +1143,7 @@ def get_datasets(
     region: str = "USA",
     delay: int = 1,
     universe: str = "TOP3000",
-    theme: str = "false",
+    theme: str = "ALL",
 ) -> pd.DataFrame:
     """
     Retrieve available datasets based on specified parameters.
@@ -1154,18 +1154,39 @@ def get_datasets(
         region (str, optional): The region. Defaults to "USA".
         delay (int, optional): The delay. Defaults to 1.
         universe (str, optional): The universe. Defaults to "TOP3000".
-        theme (str, optional): The theme. Defaults to "false".
+        theme (str, optional): The theme. Defaults to "ALL".
 
     Returns:
         pandas.DataFrame: A DataFrame containing information about available datasets.
     """
-    url = (
-        brain_api_url
-        + "/data-sets?"
-        + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
-    )
-    result = s.get(url)
-    datasets_df = pd.DataFrame(result.json()["results"])
+    if theme == "ALL":
+        # Fetch both theme=false and theme=true
+        url_false = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=false"
+        )
+        result_false = s.get(url_false)
+        df_false = pd.DataFrame(result_false.json()["results"])
+        
+        url_true = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=true"
+        )
+        result_true = s.get(url_true)
+        df_true = pd.DataFrame(result_true.json()["results"])
+        
+        datasets_df = pd.concat([df_false, df_true], ignore_index=True)
+    else:
+        url = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
+        )
+        result = s.get(url)
+        datasets_df = pd.DataFrame(result.json()["results"])
+        
     datasets_df = expand_dict_columns(datasets_df)
     return datasets_df
 

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3}/cnhkmcp/untracked/APP//347/274/230/345/210/206/344/270/200/351/201/223/346/241/245/ace_lib.py

@@ -1134,7 +1134,7 @@ def get_datasets(
     region: str = "USA",
     delay: int = 1,
     universe: str = "TOP3000",
-    theme: str = "false",
+    theme: str = "ALL",
 ) -> pd.DataFrame:
     """
     Retrieve available datasets based on specified parameters.
@@ -1145,18 +1145,39 @@ def get_datasets(
         region (str, optional): The region. Defaults to "USA".
         delay (int, optional): The delay. Defaults to 1.
         universe (str, optional): The universe. Defaults to "TOP3000".
-        theme (str, optional): The theme. Defaults to "false".
+        theme (str, optional): The theme. Defaults to "ALL".
 
     Returns:
         pandas.DataFrame: A DataFrame containing information about available datasets.
     """
-    url = (
-        brain_api_url
-        + "/data-sets?"
-        + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
-    )
-    result = s.get(url)
-    datasets_df = pd.DataFrame(result.json()["results"])
+    if theme == "ALL":
+        # Fetch both theme=false and theme=true
+        url_false = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=false"
+        )
+        result_false = s.get(url_false)
+        df_false = pd.DataFrame(result_false.json()["results"])
+        
+        url_true = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme=true"
+        )
+        result_true = s.get(url_true)
+        df_true = pd.DataFrame(result_true.json()["results"])
+        
+        datasets_df = pd.concat([df_false, df_true], ignore_index=True)
+    else:
+        url = (
+            brain_api_url
+            + "/data-sets?"
+            + f"instrumentType={instrument_type}&region={region}&delay={str(delay)}&universe={universe}&theme={theme}"
+        )
+        result = s.get(url)
+        datasets_df = pd.DataFrame(result.json()["results"])
+        
     datasets_df = expand_dict_columns(datasets_df)
     return datasets_df
 

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3}/cnhkmcp/untracked/APP//350/277/220/350/241/214/346/211/223/345/274/200/346/210/221.py

@@ -2253,7 +2253,8 @@ def inspiration_datasets():
             )
             df = df[mask]
             
-        results = df.head(50).to_dict(orient='records')
+        # Return all results instead of limiting to 50
+        results = df.to_dict(orient='records')
         return jsonify(results)
     except Exception as e:
         return jsonify({'error': str(e)}), 500

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3/cnhkmcp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 2.0.2
+Version: 2.0.3
 Summary: A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration
 Home-page: https://github.com/cnhk/cnhkmcp
 Author: CNHK

{cnhkmcp-2.0.2 → cnhkmcp-2.0.3}/setup.py

@@ -13,7 +13,7 @@ def read_requirements():
 
 setup(
     name="cnhkmcp",
-    version="2.0.2",
+    version="2.0.3",
     author="CNHK",
     author_email="cnhk@example.com",
     description="A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration",