From 94084084f3e16bbe69c5287427436fdab450befd Mon Sep 17 00:00:00 2001 From: Moonlit Jolteon Date: Wed, 23 Apr 2025 11:07:51 -0400 Subject: [PATCH] I need to remember to commit more --- config.json.example | 3 ++- data_sources/isa.py | 10 ++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/config.json.example b/config.json.example index 60a6c0c..e68b79f 100644 --- a/config.json.example +++ b/config.json.example @@ -1,5 +1,6 @@ { "TBA_TOKEN": "Get your read API token here: https://www.thebluealliance.com/account", "USE_ISA_DATA": false, - "ISA_TOKEN": "If you are a member of the Indiana Scouting Alliance, put your token here. Reach out to the discord if you don't know how to get it" + "ISA_TOKEN": "If you are a member of the Indiana Scouting Alliance, put your token here. Reach out to the discord if you don't know how to get it", + "OLLAMA_MODEL": "dolphin-mixtral:latest" } \ No newline at end of file diff --git a/data_sources/isa.py b/data_sources/isa.py index 0fc5f3d..27fd6c5 100644 --- a/data_sources/isa.py +++ b/data_sources/isa.py @@ -62,6 +62,16 @@ class IndianaScoutingAllianceConnector(DataSource): if response.status_code == 200: return response.json() + def get_robot_notes(self, team_number, event_code=None): + notes_url = self.__build_ISA_robot_url( + "0011000000000000000000000000010000000000000000000000000000000000000000000000000000000000000", + [str(team_number)], + event_code, + ) + response = requests.get(notes_url, headers=self.__headers) + if response.status_code == 200: + return response.json() + def get_team_info(self, team_number): pass