DeepSeek-R1: Testing with multiple AMD GPUs

import json
import os

def main():
    while True:
        try:
            # Get input file path
            path = input("Paste file path: ").strip()
            if path.lower() == "quit":
                break

            # Remove quotes if necessary
            if path.startswith('"') and path.endswith('"'):
                path = path[1:-1]

            # Check if file exists and has a json extension
            if not os.path.exists(path) or "json" not in os.path.splitext(path)[1]:
                print("Invalid file.")
                continue

            # Read and parse the JSON file
            with open(path, "r", encoding="utf-8") as f:
                data = json.load(f)

            # Expect the JSON to be a list with a single element
            if not isinstance(data, list) or len(data) != 1:
                print("Invalid file.")
                continue

            obj = data[0]

            # Navigate to chat -> history -> messages
            messages = obj.get("chat", {}).get("history", {}).get("messages")
            if messages is None:
                print("Invalid file.")
                continue

            # Extract usage objects from messages
            usages = []
            for key, message in messages.items():
                usage = message.get("usage")
                if usage is not None:
                    usages.append(usage)

            if not usages:
                print("No usage data found.")
                continue

            # Calculate averages
            total_time = 0.0
            total_tokens = 0.0
            valid_count = 0

            for usage in usages:
                # Get the values; if missing, skip this usage
                total_duration = usage.get("total_duration")
                resp_tokens = usage.get("response_token/s")
                if total_duration is None or resp_tokens is None:
                    continue
                total_time += total_duration
                total_tokens += resp_tokens
                valid_count += 1

            if valid_count == 0:
                print("No valid usage data found.")
                continue

            # Calculate average time (convert from nanoseconds to seconds) and tokens per second
            avg_time_secs = (total_time / valid_count) / 1_000_000_000.0
            avg_tokens = total_tokens / valid_count

            # Print the results
            print(f"Tokens/Sec: {avg_tokens:.3f}")
            print(f"Time /Secs: {avg_time_secs:.3f}")

        except Exception as ex:
            print("Error:", ex)

if __name__ == "__main__":
    main()