Skip to content

Commit

Permalink
feat: Triton integration (#2)
Browse files Browse the repository at this point in the history
* WIP

* WIP - adapt stream response

* Add triton client

* Remove test file

* WIP

* WIP fetch stream

* Fix using stream api

* Env vars

* adjust types

* Fix lockfile

* Add todos

* Fix double-query

* Display highest highlight in article-hit

* Adapt article drawer

* Fix build

* Cleanup logs & remove tests from CI

---------

Co-authored-by: Etienne Soulard-Geoffrion <etienne.soulard-geoffrion@clinia.com>
  • Loading branch information
xWiiLLz and etiennecl authored Sep 9, 2024
1 parent 0a83006 commit 739e404
Show file tree
Hide file tree
Showing 22 changed files with 1,059 additions and 190 deletions.
5 changes: 5 additions & 0 deletions .env.example
Original file line number Diff line number Diff line change
Expand Up @@ -21,3 +21,8 @@ OTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:55679
OTEL_EXPORTER_OTLP_INSECURE=true
OTEL_EXPORTER_OTLP_PROTOCOL=grpc
NEXT_OTEL_VERBOSE=0

# Inference
INFERENCE_URL=http://127.0.0.1:8001
INFERENCE_MODEL_NAME=summarizer_medical_journals_qa
INFERENCE_MODEL_VERSION=120240905190000
5 changes: 3 additions & 2 deletions .github/workflows/pr-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -39,5 +39,6 @@ jobs:
- name: Check format
run: pnpm run format:check

- name: Test
run: pnpm run test
# Bring back when we have actual tests
# - name: Test
# run: pnpm run test
13 changes: 10 additions & 3 deletions package.json
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
"version": "0.1.0",
"private": true,
"scripts": {
"dev": "next dev",
"dev": "next dev -p 3100",
"build": "next build",
"start": "next start",
"lint": "next lint",
Expand All @@ -15,8 +15,11 @@
"@auth0/nextjs-auth0": "^3.5.0",
"@clinia-ui/icons": "^0.1.8",
"@clinia-ui/react": "^0.1.8",
"@clinia/client-common": "1.0.10-hgs-84d73454.0",
"@clinia/client-datapartition": "1.0.10-hgs-84d73454.0",
"@clinia/search-sdk-core": "^0.1.0",
"@clinia/search-sdk-react": "^0.1.0",
"@clinia/tritonclient": "^1.0.0",
"@uidotdev/usehooks": "^2.4.1",
"@vercel/otel": "^1.5.0",
"clsx": "^2.1.0",
Expand All @@ -29,7 +32,9 @@
"react-dom": "^18",
"react-hook-form": "7.48.2",
"react-markdown": "^9.0.1",
"tailwind-merge": "^2.2.1"
"sanitize-html": "^2.13.0",
"tailwind-merge": "^2.2.1",
"uuid": "^10.0.0"
},
"devDependencies": {
"@testing-library/react": "^14.2.1",
Expand All @@ -38,6 +43,8 @@
"@types/node": "^20",
"@types/react": "^18",
"@types/react-dom": "^18",
"@types/sanitize-html": "^2.13.0",
"@types/uuid": "^10.0.0",
"@vitejs/plugin-react": "^4.2.1",
"autoprefixer": "^10.0.1",
"eslint": "^8",
Expand All @@ -52,4 +59,4 @@
"typescript": "^5",
"vitest": "^1.3.1"
}
}
}
Loading

0 comments on commit 739e404

Please sign in to comment.