PROTHAM commited on
Commit
4fb7a60
·
1 Parent(s): 54c7f90

fix: replace Optional[dict] with Any for openenv validate

Browse files
Files changed (1) hide show
  1. models.py +7 -4
models.py CHANGED
@@ -1,17 +1,20 @@
1
- from openenv.core.env_server import Environment
2
  try:
3
  from openenv.core.env_server.types import Action, Observation, State
4
  except ImportError:
5
- from openenv.core.models import Action, Observation, State
 
 
 
 
6
  from pydantic import Field
7
- from typing import Optional
8
 
9
 
10
  class APIAction(Action):
11
  method: str = Field(default="GET")
12
  url: str = Field(default="/mock_api/users")
13
  headers: dict = Field(default_factory=dict)
14
- body: Optional[dict] = Field(default=None)
15
  query_params: dict = Field(default_factory=dict)
16
 
17
 
 
 
1
  try:
2
  from openenv.core.env_server.types import Action, Observation, State
3
  except ImportError:
4
+ try:
5
+ from openenv.core.models import Action, Observation, State
6
+ except ImportError:
7
+ from openenv.core.env_server import Action, Observation, State
8
+
9
  from pydantic import Field
10
+ from typing import Any
11
 
12
 
13
  class APIAction(Action):
14
  method: str = Field(default="GET")
15
  url: str = Field(default="/mock_api/users")
16
  headers: dict = Field(default_factory=dict)
17
+ body: Any = Field(default=None)
18
  query_params: dict = Field(default_factory=dict)
19
 
20