Rakshithn123 commited on
Commit
02ee58e
·
1 Parent(s): 94ddaba

added hf web inteface

Browse files
Files changed (1) hide show
  1. server/app.py +4 -10
server/app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  from functools import partial
3
  from fastapi import FastAPI
4
- from openenv.core.env_server import create_app
5
  from server.second_brain_env_environment import SecondBrainEnvironment
6
  from models import SecondBrainAction, SecondBrainObservation
7
 
@@ -15,19 +15,13 @@ _task = os.getenv("TASK_NAME", "note_categorization")
15
  # Use partial to bake task_name into the constructor
16
  TaskEnv = partial(SecondBrainEnvironment, task_name=_task)
17
 
18
- app = create_app(
19
- TaskEnv,
20
  action_cls=SecondBrainAction,
21
  observation_cls=SecondBrainObservation,
 
22
  )
23
 
24
- @app.get("/")
25
- def home():
26
- return {
27
- "message": "Second Brain OpenEnv is online 🚀",
28
- "task": _task,
29
- }
30
-
31
  @app.get("/tasks")
32
  def list_tasks():
33
  return {
 
1
  import os
2
  from functools import partial
3
  from fastapi import FastAPI
4
+ from openenv.core.env_server import create_web_interface_app
5
  from server.second_brain_env_environment import SecondBrainEnvironment
6
  from models import SecondBrainAction, SecondBrainObservation
7
 
 
15
  # Use partial to bake task_name into the constructor
16
  TaskEnv = partial(SecondBrainEnvironment, task_name=_task)
17
 
18
+ app = create_web_interface_app(
19
+ env=TaskEnv,
20
  action_cls=SecondBrainAction,
21
  observation_cls=SecondBrainObservation,
22
+ env_name=_task,
23
  )
24
 
 
 
 
 
 
 
 
25
  @app.get("/tasks")
26
  def list_tasks():
27
  return {