7beshoyarnest commited on
Commit
3103e1a
·
1 Parent(s): a85ee3f

fix IPython import for production

Browse files
Files changed (2) hide show
  1. =1.26.4 +0 -0
  2. src/agent_graph/agent_backend.py +5 -1
=1.26.4 ADDED
File without changes
src/agent_graph/agent_backend.py CHANGED
@@ -1,10 +1,14 @@
1
  import json
2
- from IPython.display import Image, display
3
  from typing import Annotated, Literal
4
  from typing_extensions import TypedDict
5
  from langchain_core.messages import ToolMessage
6
  from langgraph.graph.message import add_messages
7
 
 
 
 
 
 
8
 
9
  class State(TypedDict):
10
  """Represents the state structure containing a list of messages.
 
1
  import json
 
2
  from typing import Annotated, Literal
3
  from typing_extensions import TypedDict
4
  from langchain_core.messages import ToolMessage
5
  from langgraph.graph.message import add_messages
6
 
7
+ try:
8
+ from IPython.display import Image, display
9
+ except ImportError:
10
+ pass
11
+
12
 
13
  class State(TypedDict):
14
  """Represents the state structure containing a list of messages.