Increase request timeout and fix pylint issue

devchat-ai · Jun 6, 2023 · 2090a06 · 2090a06
1 parent 17e47ac
commit 2090a06
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/devchat/openai/openai_chat.py b/devchat/openai/openai_chat.py
@@ -22,7 +22,7 @@ class OpenAIChatConfig(BaseModel):
     frequency_penalty: Optional[float] = Field(None, ge=-2.0, le=2.0)
     logit_bias: Optional[Dict[int, float]] = Field(None)
     user: Optional[str] = Field(None)
-    request_timeout: Optional[int] = Field(15, ge=3)
+    request_timeout: Optional[int] = Field(20, ge=3)
 
     class Config:
         """

diff --git a/devchat/store.py b/devchat/store.py
@@ -146,7 +146,8 @@ def select_prompts(self, start: int, end: int, topic: Optional[str] = None) -> L
         """
         if topic:
             ancestors = nx.ancestors(self._graph, topic)
-            nodes_with_data = [(node, self._graph.nodes[node]) for node in ancestors] + [(topic, self._graph.nodes[topic])]
+            nodes_with_data = [(node, self._graph.nodes[node]) for node in ancestors] + \
+                [(topic, self._graph.nodes[topic])]
             sorted_nodes = sorted(nodes_with_data, key=lambda x: x[1]['timestamp'], reverse=True)
         else:
             sorted_nodes = sorted(self._graph.nodes(data=True),