New Argument Parsing

pyproject.toml

@@ -45,6 +45,11 @@ dependencies = [
     'uvicorn[standard] == 0.30.6',
     'dataclasses_json == 0.6.7',
     'websockets == 13.1',
+    'langchain-community',
+    'langchain-openai',
+    'markdown',
+    'chromadb',
+    'langchain-chroma',
 ]
 
 [project.urls]

src/hackingBuddyGPT/utils/__init__.py

@@ -1,4 +1,4 @@
-from .configurable import Configurable, configurable
+from .configurable import Configurable, configurable, parameter
 from .console import *
 from .db_storage import *
 from .llm_util import *

src/hackingBuddyGPT/utils/configurable.py

@@ -2,33 +2,64 @@
 import dataclasses
 import inspect
 import os
-from dataclasses import dataclass
+from dataclasses import dataclass, Field, MISSING, _MISSING_TYPE
 from types import NoneType
-from typing import Any, Dict, Type, TypeVar, Set, Union
+from typing import Any, Dict, Type, TypeVar, Set, Union, Optional, overload
 
 from dotenv import load_dotenv
 
 load_dotenv()
 
 
+T = TypeVar("T")
+
+
+@overload
+def parameter(
+    *,
+    desc: str,
+    default: T = ...,
+    init: bool = True,
+    repr: bool = True,
+    hash: Optional[bool] = None,
+    compare: bool = True,
+    metadata: Optional[Dict[str, Any]] = ...,
+    kw_only: Union[bool, _MISSING_TYPE] = MISSING,
+) -> T:
+    ...
+
+@overload
 def parameter(
     *,
     desc: str,
-    default=dataclasses.MISSING,
+    default: T = ...,
     init: bool = True,
     repr: bool = True,
-    hash=None,
+    hash: Optional[bool] = None,
     compare: bool = True,
-    metadata: Dict = None,
-    kw_only: bool = dataclasses.MISSING,
-):
+    metadata: Optional[Dict[str, Any]] = ...,
+    kw_only: Union[bool, _MISSING_TYPE] = MISSING,
+) -> Field[T]:
+    ...
+
+def parameter(
+    *,
+    desc: str,
+    default: T = MISSING,
+    init: bool = True,
+    repr: bool = True,
+    hash: Optional[bool] = None,
+    compare: bool = True,
+    metadata: Optional[Dict[str, Any]] = None,
+    kw_only: Union[bool, _MISSING_TYPE] = MISSING,
+) -> Field[T]:
     if metadata is None:
         metadata = dict()
     metadata["desc"] = desc
 
     return dataclasses.field(
         default=default,
-        default_factory=dataclasses.MISSING,
+        default_factory=MISSING,
         init=init,
         repr=repr,
         hash=hash,
@@ -109,7 +140,7 @@ def create():
                 instance = self.type(**args)
                 if hasattr(instance, "init") and not getattr(self.type, "__transparent__", False):
                     instance.init()
-                setattr(instance, "configurable_recreate", create)
+                setattr(instance, "configurable_recreate", create)  # noqa: B010
                 return instance
 
             return create()
@@ -228,9 +259,6 @@ def inner(cls) -> Configurable:
     return inner
 
 
-T = TypeVar("T")
-
-
 def Global(subclass: T, global_name: str = None) -> T:
     class Cloned(subclass):
         __global__ = True