File tree 2 files changed +11
-2
lines changed
2 files changed +11
-2
lines changed Original file line number Diff line number Diff line change 3
3
4
4
import os
5
5
6
- import query
7
6
import ray
8
7
import requests
9
8
from fastapi import FastAPI
12
11
from slack_bolt import App
13
12
from slack_bolt .adapter .socket_mode import SocketModeHandler
14
13
14
+ from app import query
15
15
from app .config import (
16
16
DB_CONNECTION_STRING ,
17
17
OPENAI_API_KEY ,
@@ -69,7 +69,10 @@ class Answer(BaseModel):
69
69
@serve .ingress (app )
70
70
class RayAssistantDeployment :
71
71
def __init__ (self ):
72
- self .agent = query .QueryAgent ()
72
+ self .agent = query .QueryAgent (
73
+ llm = "meta-llama/Llama-2-70b-chat-hf" ,
74
+ max_context_length = 4096 ,
75
+ )
73
76
self .app = SlackApp .remote ()
74
77
# Run the Slack app in the background
75
78
self .runner = self .app .run .remote ()
Original file line number Diff line number Diff line change
1
+ name : " ray-assistant"
2
+ cluster_env : ray-assistant
3
+ ray_serve_config :
4
+ import_path : app.serve:deployment
5
+ runtime_env :
6
+ working_dir : " https://github.com/ray-project/llm-applications/archive/refs/tags/v0.0.1.zip"
You can’t perform that action at this time.
0 commit comments