fix: import error when exploring tables in pipelines that call segmen…

…ts (#1074)

### Summary of Changes

Exploring a table in the following code

```
package bug

pipeline whoSurvived {
    val titanic = Table.fromCsvFile("titanic.csv");

    val beforeSplit = preprocessBeforeSplit(titanic);
}

segment preprocessBeforeSplit(table: Table) -> result: Table {
    yield result = table;
}
```

lead to a runtime error `cannot import name 'preprocessBeforeSplit' from
'bug.gen_titanic' (unknown location)`.

The document that was created for profiling was not properly built, so
an incorrect import was created for the segment, even though it was in
the same document.

packages/safe-ds-lang/src/language/index.ts

@@ -20,10 +20,15 @@ export { locationToString, positionToString, rangeToString } from '../helpers/lo
 // Messages
 export * as messages from './runtime/messages.js';
 
-// Constants
+// Commands
 export * as commands from './communication/commands.js';
+
+// RPC
 export * as rpc from './communication/rpc.js';
 
+// Generation
+export { CODEGEN_PREFIX } from './generation/safe-ds-python-generator.js';
+
 // Dependencies
 export const dependencies = {
     'safe-ds-runner': {

packages/safe-ds-lang/tests/language/lsp/safe-ds-rename-provider.test.ts

@@ -9,7 +9,6 @@ import { TextDocument } from 'vscode-languageserver-textdocument';
 const services = (await createSafeDsServices(NodeFileSystem)).SafeDs;
 const documentBuilder = services.shared.workspace.DocumentBuilder;
 const langiumDocuments = services.shared.workspace.LangiumDocuments;
-const langiumDocumentFactory = services.shared.workspace.LangiumDocumentFactory;
 const nameProvider = services.references.NameProvider;
 const renameProvider = services.lsp.RenameProvider!;
 
@@ -258,11 +257,9 @@ describe('SafeDsRenameProvider', async () => {
         }
 
         // Add documents to workspace
-        const documents = descriptions.map((description) => {
-            const document = langiumDocumentFactory.fromString(description.originalContent, URI.parse(description.uri));
-            langiumDocuments.addDocument(document);
-            return document;
-        });
+        const documents = descriptions.map((description) =>
+            langiumDocuments.createDocument(URI.parse(description.uri), description.originalContent),
+        );
         await documentBuilder.build(documents);
 
         // Get the element to rename

packages/safe-ds-vscode/src/extension/eda/apis/runnerApi.ts

@@ -1,18 +1,17 @@
 import { Base64Image, Column, Profiling, ProfilingDetailStatistical, Table } from '@safe-ds/eda/types/state.js';
-import { ast, messages, SafeDsServices } from '@safe-ds/lang';
-import { AstNode, LangiumDocument } from 'langium';
+import { ast, CODEGEN_PREFIX, messages, SafeDsServices } from '@safe-ds/lang';
+import { LangiumDocument } from 'langium';
 import * as vscode from 'vscode';
 import crypto from 'crypto';
 import { getPipelineDocument } from '../../mainClient.ts';
-import { CODEGEN_PREFIX } from '../../../../../safe-ds-lang/src/language/generation/safe-ds-python-generator.ts';
 import { safeDsLogger } from '../../helpers/logging.js';
 
 export class RunnerApi {
     services: SafeDsServices;
     pipelinePath: vscode.Uri;
     pipelineName: string;
     pipelineNode: ast.SdsPipeline;
-    baseDocument: LangiumDocument<AstNode> | undefined;
+    baseDocument: LangiumDocument | undefined;
     placeholderCounter = 0;
 
     constructor(
@@ -52,12 +51,18 @@ export class RunnerApi {
             const afterPipelineEnd = documentText.substring(endOfPipeline - 1);
             const newDocumentText = beforePipelineEnd + addedLines + afterPipelineEnd;
 
-            const newDoc = this.services.shared.workspace.LangiumDocumentFactory.fromString(
-                newDocumentText,
+            this.services.shared.workspace.LangiumDocuments.deleteDocument(this.pipelinePath);
+            const newDoc = this.services.shared.workspace.LangiumDocuments.createDocument(
                 this.pipelinePath,
+                newDocumentText,
             );
+            await this.services.shared.workspace.DocumentBuilder.build([newDoc]);
+
             await this.services.runtime.Runner.executePipeline(pipelineExecutionId, newDoc, this.pipelineName);
 
+            this.services.shared.workspace.LangiumDocuments.deleteDocument(this.pipelinePath);
+            this.services.shared.workspace.LangiumDocuments.addDocument(this.baseDocument);
+
             const runtimeCallback = (message: messages.RuntimeProgressMessage) => {
                 if (message.id !== pipelineExecutionId) {
                     return;