Skip to content

Commit

Permalink
Allow running on remote orchestrator
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexejPenner committed Oct 30, 2023
1 parent 16c49fc commit 7a6600b
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 4 deletions.
3 changes: 2 additions & 1 deletion langchain-llamaindex-slackbot/src/local_testing_slackbot.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def reply_in_thread(body: dict, say, context):
thread_ts = event.get("thread_ts", None) or event["ts"]

if context["bot_user_id"] in event["text"]:
logger.debug(f"Received message: {event['text']}")
logger.info(f"Received message: {event['text']}")
if event.get("thread_ts", None):
full_thread = [
f"{msg['text']}"
Expand Down Expand Up @@ -107,6 +107,7 @@ def reply_in_thread(body: dict, say, context):
question=event["text"],
verbose=True,
)
logger.info(output)
say(text=output, thread_ts=thread_ts)


Expand Down
23 changes: 20 additions & 3 deletions langchain-llamaindex-slackbot/src/pipelines/index_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,17 +11,34 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
# or implied. See the License for the specific language governing
# permissions and limitations under the License.

import os

from steps.index_generator import index_generator
from steps.url_scraper import url_scraper
from steps.web_url_loader import web_url_loader
from zenml import pipeline
from zenml.config import DockerSettings

pipeline_name = "zenml_docs_index_generation"
docker_settings = DockerSettings(
requirements=[
"langchain==0.0.263",
"openai==0.27.2",
"slack-bolt==1.16.2",
"slack-sdk==3.20.0",
"fastapi",
"flask",
"uvicorn",
"gcsfs==2023.5.0",
"faiss-cpu==1.7.3",
"unstructured==0.5.7",
"tiktoken",
"bs4"
],
environment={"OPENAI_API_KEY": os.getenv("OPENAI_API_KEY")}
)


@pipeline(name=pipeline_name)
@pipeline(name=pipeline_name, settings={"docker": docker_settings})
def docs_to_index_pipeline(
docs_url: str = "",
repo_url: str = "",
Expand Down

0 comments on commit 7a6600b

Please sign in to comment.