Commit graph

17 commits

Author SHA1 Message Date
wyx-hhhh
ff39d8b6f7 feat:Cleaned up code comment 2025-05-30 17:24:46 +08:00
wyx-hhhh
fc4039169b Enhance cloud and local service management with status tracking and improved progress reporting
- Implemented service status file management in cloud and local services to track active status and model information.
- Added endpoints to start and stop cloud services, including validation for existing services.
- Enhanced local service management with status checks and progress updates during document processing and chunk embedding.
- Introduced real-time progress tracking for document embedding and chunk processing, allowing for incremental updates.
- Improved error handling and logging throughout the service management processes.
- Refactored chat request handling to intelligently route between local and cloud services based on current status.
2025-05-30 15:53:37 +08:00
wyx-hhhh
5a1620dee2 feat: Enhance cloud training and inference capabilities
- Updated TrainingProgress component to handle cloud training progress data and job ID.
- Modified trainExposureModel to allow nullable path and added optional stageName.
- Enhanced useSSE hook to support cloud model inference with new parameters.
- Introduced CloudProgressData type to align cloud training progress with local training structure.
- Implemented cloud inference request handling with local knowledge retrieval in cloudService.
- Added utility functions for managing active cloud model state in cloudModelUtils.
- Updated cloud inference endpoint to support local knowledge retrieval before cloud inference.
- Refactored advanced chat service to utilize new message structure for cloud inference.
- Enhanced prompt strategies to incorporate knowledge retrieval based on user messages.
2025-05-28 11:44:21 +08:00
yanmuyuan
7e53db7965 Merge branch 'feature/CloudService' 2025-05-21 15:56:53 +08:00
wyx-hhhh
65ce8d4309 Update the model service startup logic, add integrity check for the model path, and support obtaining the model path from different fields 2025-05-21 15:32:59 +08:00
yanmuyuan
cb5d0cb4da Add gguf path to json file 2025-05-20 19:42:17 +08:00
wyx-hhhh
2c8cd6cc0e Change the data type of the is_trained field to boolean and update the related logic to reflect this change 2025-05-20 19:06:35 +08:00
wyx-hhhh
33952c1ef6 Change the data type of the is_trained field to boolean and update the related logic to reflect this change 2025-05-20 17:41:30 +08:00
yanmuyuan
94e960e1c8 Enhance GGUF model handling with timestamps, metadata and memory training status 2025-05-12 10:24:18 +08:00
yexiangle
ff9a9b9970
delete useless code in route_l2.py (#332) 2025-05-07 13:40:10 +08:00
Zachary Pitroda
053090937d
Added CUDA support (#228)
* Add CUDA support

- CUDA detection
- Memory handling
- Ollama model release after training

* Fix logging issue

added cuda support flag so log accurately reflected cuda toggle

* Update llama.cpp rebuild

Changed llama.cpp to only check if cuda support is enabled and if so rebuild during the first build rather than each run

* Improved vram management

Enabled memory pinning and optimizer state offload

* Fix CUDA check

rewrote llama.cpp rebuild logic, added manual y/n toggle if user wants to enable cuda support

* Added fast restart and fixed CUDA check command

Added make docker-restart-backend-fast to restart the backend and reflect code changes without causing a full llama.cpp rebuild

Fixed make docker-check-cuda command to correctly reflect cuda support

* Added docker-compose.gpu.yml

Added docker-compose.gpu.yml to fix error on machines without nvidia gpu and made sure "\n" is added before .env modification

* Fixed cuda toggle

Last push accidentally broke cuda toggle

* Code review fixes

Fixed errors resulting from removed code:
- Added return save_path to end of save_hf_model function
- Rolled back download_file_with_progress function

* Update Makefile

Use cuda by default when using docker-restart-backend-fast

* Minor cleanup

Removed unnecessary makefile command and fixed gpu logging

* Delete .gpu_selected

* Simplified cuda training code

- Removed dtype setting to let torch automatically handle it
- Removed vram logging
- Removed Unnecessary/old comments

* Fixed gpu/cpu selection

Made "make docker-use-gpu/cpu" command work with .gpu_selected flag and changed "make docker-restart-backend-fast" command to respect flag instead of always using gpu

* Fix Ollama embedding error

Added custom exception class for Ollama embeddings, which seemed to be returning keyword arguments while the Python exception class only accepts positional ones

* Fixed model selection & memory error

Fixed training defaulting to 0.5B model regardless of selection and fixed "free(): double free detected in tcache 2" error caused by cuda flag being passed incorrectly
2025-04-25 10:20:36 +08:00
KKKKKKKevin
0e442ed17c
Fix/make meta data field optional (#237)
* fix:metadata to optional

* docker command check

* add exception stack
2025-04-15 19:51:33 +08:00
GoForceX
9ac11344b1
fix: Find correct llama-server location (#186) 2025-04-14 13:10:49 +08:00
yuchengzhou
8741ef364e
Feature/0327/adapter chat to openai (#121)
* feat(chat): Chat Interface Protocol Modification to Align with OpenAI
2025-03-31 15:49:11 +08:00
KKKKKKKevin
27c76ae29c
Fix:Env detect & Fix Missing Merge Step (#113)
* hotfix: Fix missing final data merge & Fix Env Detect In Script Runner

---------

Co-authored-by: yingapple <yingxiang835@gmail.com>
2025-03-30 18:03:52 +08:00
omahs
ca01a9e73f
Fix typos (#60)
* fix typo

* fix typo

* fix typos

* fix typos

* fix typos

* fix typo

* fix typos

* fix typos

* fix typos
2025-03-27 10:18:39 +08:00
Kevin
7f7d64210e Initial commit 2025-03-20 00:37:54 +08:00