diff --git a/copilot/context.py b/copilot/context.py index e1f0afb..f6e55ab 100644 --- a/copilot/context.py +++ b/copilot/context.py @@ -1,4 +1,5 @@ from dataclasses import dataclass +from typing import List from copilot.conversation import Model from copilot.parse_os import OperatingSystem @@ -9,7 +10,7 @@ class Context: shell: str operating_system: OperatingSystem directory: str - directory_list: list[str] + directory_list: List[str] history: str command: str git: str diff --git a/copilot/conversation.py b/copilot/conversation.py index 1a230d4..167e9b3 100644 --- a/copilot/conversation.py +++ b/copilot/conversation.py @@ -1,6 +1,7 @@ import argparse from dataclasses import dataclass from enum import Enum +from typing import List, Dict class Model(Enum): @@ -18,5 +19,5 @@ def argparse_model_type(model_str): @dataclass class Conversation: - messages: list[dict] + messages: List[Dict] model: Model